numpy 索引疊代

import numpy as np

A = np.arange(3, 15).reshape((3,4))
print(A)
print(A[1,1])
print(A[1][1])
print('*************************')
# 列印第一行所有的數
print(A[1,:])
print('*************************')
# 列印第1行第1-2列的值
print(A[1,1:3])
print('*************************')
# 疊代行
for row in A:
    print(row)
print('*************************')
# 疊代列
for column in A.T:
    print(column)
print('*************************')
print(A.flatten())
# 疊代每個元素
for item in A.flat:
    print(item)
           

結果：

[[ 3 4 5 6]

[ 7 8 9 10]

[11 12 13 14]]

8

8

[ 7 8 9 10]

[8 9]

[3 4 5 6]

[ 7 8 9 10]

[11 12 13 14]

[ 3 7 11]

[ 4 8 12]

[ 5 9 13]

[ 6 10 14]

[ 3 4 5 6 7 8 9 10 11 12 13 14]

3

4

5

6

7

8

9

10

11

12

13

14