python - numpy.ndarray.shape 改变维度

Question

保存 numpy 数组 (numpy.ndarray.shape) 维度的元组会更改大小。例如:

len(numpy.array([1,2,3]).shape) -> 1, shape=(1,)
len(numpy.array([[1,2,3],[4,5,6]]).shape) -> 2, shape=(2,3)

是否有其他方法可以使维度与数组类型保持不变？

这是我经常遇到的问题的示例:

mat3D = np.arange(27).reshape(3,3,3)
mat2D = np.arange(9)

def processMatrix(mat):
  if M.ndim == 2:
    return foo(mat)
  else:
    return np.array([foo(mat[:,:,c]) for c in range(mat.shape[2])]) 

拥有 mat2D.shape = (3,3,1) 会将代码简化为:

def processMatrix(mat):
    return np.array([foo(mat[:,:,c]) for c in range(mat.shape[2])]) 

Answer 1

您可以使用

numpy.array([[1,2,3],[4,5,6]]).ndim