数组取整列元素方法

---

title: Python中的X[:,0]和X[:,1]

Python中的X[:,0]和X[:,1]说明

今天在学习knn算法是例子中有一句：
plt.scatter(red[:, 0], red[:, 1], 80, ‘r’, ‘^’)
不知道其中的red[:, 0]是什么意思便去了解学习。

解释

X[:,0]是numpy中数组的一种写法，表示对一个二维数组，取该二维数组第一维中的所有数据，第二维中取第0列数据，
直观来说，X[:,0]就是取所有行的第0列数据, X[:,1] 就是取所有行的第1列数据。

例子：

X = np.array([[0, 1],
              [2, 3], 
              [4, 5], 
              [6, 7], 
              [8, 9], 
              [10, 11], 
              [12, 13], 
              [14, 15], 
              [16, 17], 
              [18, 19]])
print(X[:, 0])

结果：

[0 2 4 6 8 9 10 12 14 16 18]

参考：
https://blog.csdn.net/a394268045/article/details/79104219