小编use*_*439的帖子

我已经定义了对3xN NumPy数组的操作，我想遍历数组的每一列。

我试过：

for i in range(nparray.shape[1]):

但是，如果nparray.ndim == 1，则失败。

有没有一种干净的方法来确定NumPy数组的列数，例如，1确定它是否是一个1D数组（就像 MATLAB 的size操作那样）？

否则，我已经实施了：

if nparray.ndim == 1:
    num_points = 1
else:
    num_points = nparray.shape[1]

for i in range(num_points):