1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50
| import numpy as np; import seaborn as sns import pandas as pd import matplotlib.pyplot as plt
c=[] x1=np.random.normal(1,10,100) y1=np.random.normal(1,10,100) x2=np.random.normal(12,5,100) y2=np.random.normal(12,5,100) for i in range(100): c.append([x1[i],y1[i],0]) c.append([x2[i],y2[i],1])
c=(np.matrix(c)) y=c[:,[2]] X=np.matrix(c[:,[0,1]]) bt=np.dot(X.T,X) bt=np.linalg.pinv(bt) bt=np.dot(bt,X.T)*y print(bt)
haty=X*bt print(haty) plt.plot()
print(bt)
simulate_X=np.arange(-24,24,0.5) S_X=[] for x in simulate_X: S_X.append([x,(0.5-x*bt[0,0])/bt[1,0]]) S_X=np.matrix(S_X) plt.plot(S_X[:,0],S_X[:,1])
plt.scatter(x1,y1) plt.scatter(x2,y2) plt.show()
|