我无法使用 numpy 绘制用于分类算法的数据,因为它会引发此错误 ValueError: x and y must be the same size
我的数据在data
变量看起来像这样:
[[ 34.62365962 78.02469282 0. ]
[ 30.28671077 43.89499752 0. ]
[ 35.84740877 72.90219803 0. ]
[ 60.18259939 86.3085521 1. ]
[ 79.03273605 75.34437644 1. ]
[ 45.08327748 56.31637178 0. ]
[ 61.10666454 96.51142588 1. ]
[ 75.02474557 46.55401354 1. ]]
代码:
data=np.loadtxt('ex2data1.txt',delimiter=',',dtype=None)
X = data[:, [0,1]]
y = data[:, 2]
pylab.scatter(X,y)
pylab.show()
我试图绘制这个:
最佳答案
最简单的方法是在加载时解压数据
import matplotlib.pyplot as plt
x,y,c = np.loadtxt('ex2data1.txt',delimiter=',', unpack=True)
plt.scatter(x,y,c=c)
plt.show()
显然你也可以在之后进行拆包,
import matplotlib.pyplot as plt
data = np.loadtxt('ex2data1.txt',delimiter=',')
plt.scatter(data[:,0],data[:,1],c=data[:,2])
plt.show()
关于python如何绘制分类数据,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/44603609/