A-A+
Python NumPy中数组元素的访问
1、通过索引访问
在NumPy中可以直接通过索引访问数组元素, 元素的索引从0开始。
- >>> X=np.array([[51,55],[14,19],[0,4]])
- >>> X[0]
- array([51, 55])
- >>> X[1][1]
- 19
2、使用for语句遍历数组
在NumPy中可以使用 for语句访问各个元素。
- >>> X=np.array([[51,55],[14,19],[0,4]])
- >>> for row in X:
- print(row)
- [51 55]
- [14 19]
- [0 4]
- >>> for row in X:
- for elem in row:
- print(elem)
- 51
- 55
- 14
- 19
- 0
- 4
3、 使用数组访问各个元素
- >>> X=np.array([51,55,14,19,0,4])
- >>> X[np.array([0,2,4])]
- array([51, 14, 0])
4、通过布尔标记访问元素
在NumPy中可以通过布尔标记获取满足一定条件的元素。例如,要从 X中抽出大于15的元素,可以写成如下形式。
- >>> X=np.array([51,55,14,19,0,4])
- >>> X[X>15]
- array([51, 55, 19])
对NumPy数组使用不等号运算符等(上例中是 X > 15),结果会得到一个布尔型的数组。上例中就是使用这个布尔型数组取出了数组的各个元素(取出 True对应的元素)。
- >>> X=np.array([51,55,14,19,0,4])
- >>> X>15
- array([ True, True, False, True, False, False])