Question

我有这段代码：

import numpy as np

a = [0,np.array([1,2])]

try:
    # something like that
    [ x =  np.array([x]) for x in a if x == 0]
except ValueError:
    pass

我想将每个零值替换为数组，因此我的结果将是：

a = [np.array([0]), np.array([1,2])

Answer 1

您的列表理解无效。

使用：

x = [np.array([x]) for x in a if x == 0]

代替。

请注意，此处的逻辑不提供相关答案，而是：

x = array([1, 2])

最后。

对于你的期望：

使用：

np.array([0]) if x is 0 else x for x in a

将值转换为数组

1 个答案: