代码:
import numpy as np
A = np.arange(3, 15).reshape((3, 4))
print(A)
# 索引到某一行
print(A[2])
# 索引某个元素
print(A[1][1])
print(A[1, 1])
print(A[1, :]) # 第一行所有元素
print(A[:, 1]) # 第一列的所有元素
运行结果:
[[ 3 4 5 6]
[ 7 8 9 10]
[11 12 13 14]]
[11 12 13 14]
8
8
[ 7 8 9 10]
[ 4 8 12]
代码:
# # 循环
# 输出每一行
for row in A:
print("每一行")
print(row)
运行结果:
每一行
[3 4 5 6]
每一行
[ 7 8 9 10]
每一行
[11 12 13 14]
代码:
# 输出每一列
for column in A.T:
print(column)
运行结果:
[ 3 7 11]
[ 4 8 12]
[ 5 9 13]
[ 6 10 14]
代码:
# 输出每一个元素
print(A.flatten())
for item in A.flat:
print(item)
运行结果:
[ 3 4 5 6 7 8 9 10 11 12 13 14]
3
4
5
6
7
8
9
10
11
12
13
14