从np中排除零.Linspace在一行

Question

如何从linspace中删除零?

这是我的清单:

np.linspace(-3, 3, 1)
--> [-3, -2, -1, 0, 1, 2, 3] 

我想实现这个:

[x if x != 0 for x in np.linspace(-3, 3, 1)]
--> [-3, -2, -1, 1, 2, 3]

Answer 1

首先，要获得预期的结果，您需要np.linspace(-3, 3, 7)获得7个数字。其次，你的if语句必须在列表推导式的交互之后。工作代码为:

import numpy as np
print([int(x) for x in np.linspace(-3, 3, 7) if x != 0])

Answer 2

不用代码就能实现的方法。

import numpy as np
lin_s = list(map(int, np.linspace(-3, 3, 7)))
lin_s.remove(0)