python - 如何生成二维 numpy 数组？

Question

我正在尝试在生成器的帮助下生成一个 2d numpy 数组:

x = [[f(a) for a in g(b)] for b in c]

如果我尝试做这样的事情:

x = np.array([np.array([f(a) for a in g(b)]) for b in c])

如预期的那样，我得到了 np.array 的 np.array。但我想要的不是这个，而是 ndarray，所以我可以通过这样的方式获得列:

y = x[:, 1]

所以，我很好奇是否有办法以这种方式生成它。

当然可以创建所需大小的 npdarray 并用所需的值填充它，但我想要一种在一行代码中完成此操作的方法。

Answer 1

这个有效:

a = [[1, 2, 3], [4, 5, 6]]
nd_a = np.array(a)

所以这也应该有效:

nd_a = np.array([[x for x in y] for y in a])