检测 numpy 数组的维数不是形状

问题描述

获取两个不同的 numpy 数组的形状返回元组

a.shape
Out[131]: (3,)

A.shape
Out[132]: (3,3)

基于元组，一个是一维数组（维数=1），另一个是2d。如何检测类似于 type(A) 会告诉我其中一个是 numpy.ndarray 的维度数量？我应该只使用 len(a.shape) 吗？

解决方法

您应该使用 numpy.ndarray.ndim。所以

a.ndim # gives 1

和

A.ndim # gives 2