python - 如果键的值为真，则过滤字典

Question

我是 Python 的新手(第 1 天)。我有一个数据集，它通过 bool “poi”键指示某人是否是感兴趣的人。我能够使用以下内容过滤数据:

filtered = []
for n in enron_data:
    if enron_data[n]['poi']: filtered.append(enron_data[n]);
print(len(filtered))

我尝试了一段时间使用内置过滤器的 pythons 但无法使用。使用内置过滤器执行此操作的干净方法是什么？

示例数据:{'METTS MARK': {... 'poi': False,}, ...}

Answer 1

您可以使用列表推导式遍历字典，然后创建一个新的值列表，这些值对 value['poi'] 求值为 True。

filtered = [v for k, v in enron_data.items() if v['poi']]

事实上，您根本没有使用这些键，所以您可以这样做:

filtered = [v for v in enron_data.values() if v['poi']]

或者使用过滤器(类似于@AbidHasan):

filtered = filter(lambda x: x['poi'], enron_data.values())