In [9]:
from sklearn.datasets import load_iris
In [10]:
from sklearn.model_selection import train_test_split
In [11]:
import pandas as pd
In [12]:
import matplotlib
In [13]:
iris_dataset = load_iris()
In [14]:
X_train,X_test,y_train,y_test = train_test_split(iris_dataset['data'],iris_dataset['target'],random_state=0)
In [15]:
iris_dataframe = pd.DataFrame(X_train,columns = iris_dataset.feature_names)
In [16]:
print (iris_dataframe[:10])
   sepal length (cm)  sepal width (cm)  petal length (cm)  petal width (cm)
0                5.9               3.0                4.2               1.5
1                5.8               2.6                4.0               1.2
2                6.8               3.0                5.5               2.1
3                4.7               3.2                1.3               0.2
4                6.9               3.1                5.1               2.3
5                5.0               3.5                1.6               0.6
6                5.4               3.7                1.5               0.2
7                5.0               2.0                3.5               1.0
8                6.5               3.0                5.5               1.8
9                6.7               3.3                5.7               2.5
In [20]:
grr = pd.plotting.scatter_matrix(iris_dataframe,c = y_train,figsize=(15,15),marker = 'o',hist_kwds = {'bins':20},s = 60,alpha = .8)