Numpy 获取数组的行和列

代码复现：

1.  
   array_3 = np.random.randint(10, size=(3, 4, 5)) # 三维数组
2.  
   print(array_3)
3.  
   # 获取数组的行和列
4.  
   print("第一列")
5.  
   print(array_3[:, :, 0]) # 第一列
6.  
   print("第一行")
7.  
   print(array_3[:, :1, :]) # 第一行

运行结果：

1.  
   [[[5 0 3 3 7]
2.  
   [9 3 5 2 4]
3.  
   [7 6 8 8 1]
4.  
   [6 7 7 8 1]]
5.  
    
6.  
   [[5 9 8 9 4]
7.  
   [3 0 3 5 0]
8.  
   [2 3 8 1 3]
9.  
   [3 3 7 0 1]]
10.  
     
11.  
    [[9 9 0 4 7]
12.  
    [3 2 7 2 0]
13.  
    [0 4 5 5 6]
14.  
    [8 4 1 4 9]]]
15.  
    第一列
16.  
    [[5 9 7 6]
17.  
    [5 3 2 3]
18.  
    [9 3 0 8]]
19.  
    第一行
20.  
    [[[5 0 3 3 7]]
21.  
     
22.  
    [[5 9 8 9 4]]
23.  
     
24.  
    [[9 9 0 4 7]]]
25.  
     
26.  
    Process finished with exit code 0

这篇好文章是转载于：学新通技术网