0
点赞
收藏
分享

微信扫一扫

Numpy 获取数组的行和列


        一种常见的需求是获取数组的单行和单列。你可以将索引与切片组合起来实现这个功能,用一个冒号(: )表示空切片



代码复现:

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

举报

相关推荐

0 条评论