【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)