函数获取NumPy数组中的列数，如果是1D数组，则返回1

Question

我已经在3xNNumPy数组上定义了操作，我想在数组的每一列上循环。

我试过：

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

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

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

否则，我已实施：

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

for i in range(num_points):