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