一种常见的需求是获取数组的单行和单列。你可以将索引与切片组合起来实现这个功能,用一个冒号(: )表示空切片
代码复现:
array_3 = np.random.randint(10, size=(3, 4, 5)) # 三维数组
print(array_3)
# 获取数组的行和列
print("第一列")
print(array_3[:, :, 0]) # 第一列
print("第一行")
print(array_3[:, :1, :]) # 第一行
运行结果:
[[[5 0 3 3 7]
[9 3 5 2 4]
[7 6 8 8 1]
[6 7 7 8 1]]
[[5 9 8 9 4]
[3 0 3 5 0]
[2 3 8 1 3]
[3 3 7 0 1]]
[[9 9 0 4 7]
[3 2 7 2 0]
[0 4 5 5 6]
[8 4 1 4 9]]]
第一列
[[5 9 7 6]
[5 3 2 3]
[9 3 0 8]]
第一行
[[[5 0 3 3 7]]
[[5 9 8 9 4]]
[[9 9 0 4 7]]]
Process finished with exit code 0