Question

我想使用列表推导根据字典中的值创建一个0和1的向量。

在此示例中，我希望将每个正数都返回为1，而将每个0数保持为0。但是，我需要更改解决方案，以便将阈值设置为0.25（而不是0）。轻松进行更改。

test_dict = {'a':0.6, 'b':0, 'c':1, 'd':0.5}
skill_vector = [1 for skill.values() in test_dict if skill.values > 0 else 0]

所需的输出： [1,0,1,1]

编辑：明智的人指出，字典没有顺序，因此输出将不可用。有鉴于此，我打算使用OrderedDict子类。

Answer 1

您可以将测试中的布尔值转换为整数，而不是使用if/else模式：

test_dict = {'a':0.6, 'b':0, 'c':1, 'd':0.5}

threshold = 0
[int(v > threshold) for v in test_dict.values()]
# [1, 0, 1, 1]

这假定您使用的是python版本，该版本将密钥保持在插入顺序中。

Answer 2

您可以使用三元运算符：

Python 3.6.7 (default, Oct 22 2018, 11:32:17) 
[GCC 8.2.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> test_dict = {'a':0.6, 'b':0, 'c':1, 'd':0.5}
>>> [1 if x > 0 else 0 for x in test_dict.values()]
[1, 0, 1, 1]

您还可以使用字典理解来确保将结果映射到正确的键：

>>> {k:1 if v > 0 else 0 for k,v in test_dict.items()}
{'a': 1, 'b': 0, 'c': 1, 'd': 1}

Answer 3

如果要使用skill_values函数：

def skill_values(x):
     return skill_values >= .25

skill_vector = [1 if skill_values(x) else 0 for x in test_dict.values()]

或从另一个答案合并到int的映射

skill_vector = [int(skill_values(x)) for x in test_dict.values()]

Answer 4

代码：

test_dict = {'a':0.6, 'b':0, 'c':1, 'd':0.5}
skill_vector = list(map(int, map(bool, test_dict.values())))

输出：

[1, 0, 1, 1]

字典值的列表理解，根据值分配0或1

4 个答案: