获取numpy数组列的简单方法答案

【问题标题】：simple way of getting columns for numpy array获取numpy数组列的简单方法
【发布时间】：2020-10-13 19:43:07
【问题描述】：

我有一些对象y。
如果directed，y 是形状为 (n, d) 数组的元组 (y0, y1)。
如果not directed，y 是一个形状为 (n, d) 的数组。

此外，np.ndarray 类型的 y/y0/y1 可以是 1d 或 2d。如果是 1d，我想要它的长度。如果是 2d，我想要它的形状的第二个值。

我试图想出最简单的代码来解释所有这些并给我d 的值。到目前为止，我拥有的最简单的是

_, d = np.atleast_2d(y[0]).shape if directed else np.atleast_2d(y).shape

但这感觉有点迟钝。谁能想到更好的方法？

【问题讨论】：

【解决方案1】：

除非您对允许使用的行数有预算，否则这样的事情既好又简单（您甚至可以添加解释性 cmets 以获得奖励积分）：

if directed:
    [y0, y1] = y
    d = y0.shape[-1]
    assert y1.shape[-1] == d
else:
    d = y.shape[-1]

【讨论】：