Np阵列形状(1,X,X)至(X,X)

时间:2018-12-07 14:11:04

标签: python numpy numpy-ndarray

是否可以整形和np数组并避免不相关的维度1的第一个列表?

array.shape  (1,X,X)
to
array2.shape (X,X)

3 个答案:

答案 0 :(得分:2)

array = array.reshape(X,X)

-或-

array = array[0]

答案 1 :(得分:0)

只需替换形状:

write

答案 2 :(得分:0)

您可以使用squeeze方法从数组中删除尺寸为1的轴。

array = np.random.randn(1, 10, 10) 
array = array.squeeze() 
print(array.shape)
# (10, 10)