【Python之numpy库】13.取数组某一部分时保留维度

如果解决了你的问题，点个赞再走嘛٩(๑❛ᴗ❛๑)۶

当我们在对二维数组进行操作时，想取某行或某列，但是取出来后会自动变为一维。如果想保留二维维度，该怎么办呢？

方法就是加None，详细见代码

import numpy as np

a = np.array([[1, 2, 3],
              [4, 5, 6],
              [7, 8, 9],
              [1, 2, 3]])

print('a=\n', a)

print('\na[:, 2]=\n', a[:, 2])
print('\na[:, 2, None]=\n', a[:, 2, None])

print('\na[2, :]=\n', a[2, :])
print('\na[2, None, :]=\n', a[2, None, :])

结果：

a=
 [[1 2 3]
 [4 5 6]
 [7 8 9]
 [1 2 3]]

a[:, 2]=
 [3 6 9 3]

a[:, 2, None]=
 [[3]
 [6]
 [9]
 [3]]

a[2, :]=
 [7 8 9]

a[2, None, :]=
 [[7 8 9]]