%matplotlib inline
import matplotlib.pyplot as plt

import scipy.misc

w = scipy.misc.imread("HendrixCollegeMainEntrance.PNG")

w

z = w.reshape((-1, 3))

z

from scipy.cluster.vq import kmeans, vq
centroids, distortion = kmeans(z, 3)

centroids

idx, _ = vq(z, centroids)

idx

res = centroids[idx]

res

res2 = res.reshape((w.shape))

res2

scipy.misc.imsave("res2.png", res2)

import numpy as np

d = []
fin = open("FEV.csv")
fin.readline()
for line in fin.readlines():
    d.append([float(x) for x in line.split(",")[1:4]])
fin.close()

data = np.array(d)

data

plt.plot(data[:,0], data[:,1], "o")

plt.plot(data[:,0], data[:,2], "o")

plt.plot(data[:,2], data[:,1], "o")

from mpl_toolkits.mplot3d import Axes3D
fig = plt.figure()
ax = fig.add_subplot(111, projection="3d")
plt.plot(m.Y[:,0], m.Y[:,1], m.Y[:,2], "o")

from matplotlib.mlab import PCA

m = PCA(data)

dir(m)

m.a

plt.plot(m.a[:,0], m.a[:,1], "o")

m.Wt

m.fracs

m.Y

plt.plot(m.Y[:,0], m.Y[:,1], "o")

m.project([8, 4, 60])

from scipy.cluster.vq import kmeans, vq
centroids, distortion = kmeans(data, 3)

idx, _ = vq(data, centroids)

plt.plot(m.Y[idx==0,0], m.Y[idx==0,1], "o")
plt.plot(m.Y[idx==1,0], m.Y[idx==1,1], "o")
plt.plot(m.Y[idx==2,0], m.Y[idx==2,1], "o")