python - 如何构建一个以 np.nan 作为第一项的 numpy 数组

Question

所以我试图找出是否有一种方法可以做到这一点，类似于在 np.array 中使用 for 循环。下面设置了一个简单的例子。

普通方法:

>>> np.array([
... (x, x + 2)
... for x in range(10)
... if x%2 == 0])
array([[ 0,  2],
       [ 2,  4],
       [ 4,  6],
       [ 6,  8],
       [ 8, 10]])

我想要实现的输出是:

array([[np.nan, np.nan],
       [ 0,  2],
       [ 2,  4],
       [ 4,  6],
       [ 6,  8],
       [ 8, 10]])

我试过没有成功的是:

>>> np.array([
... (np.nan, np.nan),
... (x, x+2)
... for x in range(10)
... if x%2 == 0])

还有:

>>> np.array(
... [np.nan, np.nan],
... [(x, x+2)
... for x in range(10)
... if x%2 == 0])

我想在不连接 hstack 的情况下执行此操作。

非常感谢任何帮助。

Answer 1

更新以匹配更新后的问题。

>>> np.array([[np.nan, np.nan], *[(x, x+2) for x in range(10) if x%2 ==0]])
array([[nan, nan],
       [ 0.,  2.],
       [ 2.,  4.],
       [ 4.,  6.],
       [ 6.,  8.],
       [ 8., 10.]])

更新结束。

我认为你正在尝试做的是

np.array([[np.nan, np.nan], *5*[[1, 2]]])

你也可以重复 numpy-side

np.repeat([[np.nan,np.nan],[1,2]], [1,5], axis=0)