2 import matplotlib.pyplot as plt
3 IMAGEDIR = "report/pics/"
5 def generateHistograms(X, header):
7 for i, c in enumerate(X.T):
9 plt.savefig(IMAGEDIR + header[i] + "-hist.pdf")
12 def generateScatterPlot(X, Y, header):
14 for i, feature in enumerate(X.T):
15 values = np.unique(feature)
16 values = np.sort(values)
19 a =Y[np.where(feature == value)]
20 a = np.array(map(lambda x : x**(1.0/len(a)), a))
21 geomean.append(a.prod())
22 plt.plot(feature, Y, 'r.')
23 for ii in range(0, len(geomean)-1):
24 plt.plot(values[ii:ii + 2], geomean[ii:ii + 2], 'bo-')
25 plt.savefig(IMAGEDIR + header[i] + "-scat.pdf")
28 def plot(data, header):
34 X = X.astype(np.float)
36 Y = Y.astype(np.float)
37 generateHistograms(X, header)
38 generateScatterPlot(X, Y, header)