In [1]:
import pandas as pd
import matplotlib.pyplot as plt
import matplotlib
%matplotlib inline
from pandas.tools.plotting import parallel_coordinates
df = pd.read_csv('Datasets/wheat.data')
df.head(5)
Out[1]:
In [2]:
del df['id']
df.head(5)
Out[2]:
In [3]:
df.corr()
Out[3]:
In [6]:
import matplotlib.pyplot as plt
plt.imshow(df.corr(), cmap=plt.cm.Blues, interpolation='nearest')
plt.colorbar()
tick_marks = [i for i in range(len(df.columns))]
plt.xticks(tick_marks, df.columns, rotation='vertical')
plt.yticks(tick_marks, df.columns)
Out[6]: