【NumPy】索引操作

NumPy的索引跟列表索引类似

import numpy as np

array1 = np.array([1, 2, 3, 4, 5, 6])
array2 = np.array([[1, 2, 3],
                   [4, 5, 6],
                   [7, 8, 9]])
print(array1[2])  # 获取array1的第3个元素（下标从0开始），3
print(array2[2])  # 获取array2的第3行元素，[7 8 9]
print(array2[2][2])  # 获取array2的第3行第3列元素，9
print(array2[2, 2])  # 获取array2的第3行第3列元素，9
print(array2[:, 2])  # 获取array2的所有行的第3个元素，[3 6 9]

# 遍历行
for i in array2:
    print(i)
# 遍历列
for i in array2.T:
    print(i)