Python 获取numpy数组列的简单方法

我有一些对象

y

如果

定向

，则y是形状（n，d）数组的元组（y0，y1）。
如果

未定向

，则y是形状（n，d）的单个数组

此外，

np.ndarray

类型的y/y0/y1可以是1d或2d。如果是1d，我想要它的长度。如果是2d，则需要其形状的第二个值

我正试图想出最简单的代码来解释所有这些，并给出

d

的值。到目前为止，我最简单的方法是

\ud=np.至少\u2d（y[0]）。如果有指示，则为形状，否则为np.至少\u2d（y）。形状

---

但这感觉有点迟钝。有人能想出更好的方法吗？

除非您对允许使用的行数有预算，否则这样做很简单（您甚至可以为奖励积分添加解释性注释）：

如果有指示：
[y0，y1]=y
d=y0.形状[-1]
断言y1.形状[-1]==d
其他：
d=y.形状[-1]