如何生成2d 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])

Answer 2

要创建一个新数组，这似乎numpy.zeros是要走的路

import numpy as np
a = np.zeros(shape=(x, y))

您还可以设置数据类型来合理分配它

>>> np.zeros(shape=(5,2), dtype=np.uint8)
array([[0, 0],
       [0, 0],
       [0, 0],
       [0, 0],
       [0, 0]], dtype=uint8)
>>> np.zeros(shape=(5,2), dtype="datetime64[ns]")
array([['1970-01-01T00:00:00.000000000', '1970-01-01T00:00:00.000000000'],
       ['1970-01-01T00:00:00.000000000', '1970-01-01T00:00:00.000000000'],
       ['1970-01-01T00:00:00.000000000', '1970-01-01T00:00:00.000000000'],
       ['1970-01-01T00:00:00.000000000', '1970-01-01T00:00:00.000000000'],
       ['1970-01-01T00:00:00.000000000', '1970-01-01T00:00:00.000000000']],
      dtype='datetime64[ns]')

也可以看看

• 如何在 NumPy 中创建空数组/矩阵？
• np.full(size, 0) 与 np.zeros(size) 与 np.empty()