如果解决了你的问题，点个赞再走嘛٩(๑❛ᴗ❛๑)۶
当我们在对二维数组进行操作时，想取某行或某列，但是取出来后会自动变为一维。如果想保留二维维度，该怎么办呢？
方法就是加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]]