Python：为给定条件的列表元素分配布尔值

Question

假设您有以下列表：

import numpy as np

mylist=[8.4468975,
        8.52036,
        9.3605785,
        np.ma.core.MaskedConstant(),
        np.ma.core.MaskedConstant()]  

 # [8.4468975, 8.52036, 9.3605785, masked, masked]         

并且您想将0 分配给所有masked 的值，并将1 分配给所有其他值。如何正确地做到这一点？

我试过这个，但结果是一个空列表：

mylist=[1 for i,v in enumerate(mylist) if type(mylist[i]) is float]

结果应该是数字（无真/假）：

mylist=[1,1,1,0,0]

编辑

我原始列表中的masked 值属于numpy.ma.core.MaskedConstant 类型。

Answer 1

您可以使用int() 将布尔值转换为整数：

[int(isinstance(x,float)) for x in mylist]

请注意，这是因为 numpy.ma.core.MaskedConstant() 不是 float。

Answer 2

您看不到这个答案，因此尽管其他答案令人满意，但我还是给出了它。定义一个函数f

def f(x):
    if isinstance(x, np.ma.core.MaskedConstant):
        return 0
    else:
        return 1

现在做列表理解

mylist = [f(x) for x in mylist]

Answer 3

你可以在这里使用map函数：

>>> list(map(lambda x : 0 if isinstance(x,np.ma.core.MaskedConstant) else 1 , mylist))

[1,1,1,0,0]

Answer 4

像这样：

mylist=[8.4468975,
        8.52036,
        9.3605785,
        masked,
        masked]

mylist = [0 if type(x) is numpy.ma.core.MaskedConstant else 1 for x in mylist]

还请注意，您的列表理解 mylist=[1 for i,v in enumerate(mylist) if type(mylist[i]) is float] 并不完全符合您的描述；至少在一般情况下不会。您将 1 分配给不是 float 的任何东西，这与任何 'masked' 不同的东西。

还请注意，当像 [1 for i,v in enumerate(mylist) if type(mylist[i]) is float] 一样循环时，您（虽然错误，请参阅下文）不使用 v，因此您可以重写为 [1 for i,_ in enumerate(mylist) if type(mylist[i]) is float]。注意下划线。它用于暗示您将 转储enumerate() 返回的元组的第二项。

最后，我说错了，因为在您的列表理解中mylist[i] 可以替换为v。如果做得好，它会写成：[1 for v in mylist if type(v) is float]。但正如我所说，这与您似乎想要的略有不同。

Answer 5

你可以试试这个：

mylist = [1 if isinstance(v, float) else 0 for i,v in enumerate(mylist)]

或者，您可以这样做：

mylist = [0 if isinstance(v, np.ma.core.MaskedConstant) else 1 for i,v in enumerate(mylist)]

这样您就只将特定的掩码类型设置为 0。

检查变量类型的相关函数是：https://docs.python.org/2/library/functions.html#isinstance

此外，当您（正确地）使用 enumerate 构造时，您已经使用变量 v 循环遍历值，因此无需通过 mylist[i] 访问它。