筛选列表字典

Question

我有一个形式的字典：{"level"：[1,2,3]，"conf"：[-1,1,2]，"text"：["here"，"hel"，"llo"]}

我想过滤列表，以删除索引i中的每一项，其中索引i中的conf不是

输出应如下所示：

{“level”：[2,3]，“conf”：[1,2]，文本：[“hel”，“llo”]}

因为conf的第一个值不是

我试过这样的方法：

new_dict = {i: [a for a in j if a >= min_conf] for i, j in my_dict.items()}

但这只适用于一把钥匙。

回复：下面有很多好的解决方案。我只是在想另一个。

• 将这个列表结构转换成一个元组列表：元组=[（1，-1，"here"），（2,1，"hel"），（3，"2"，"llo"）]
• 然后我只需要通过每个元组的第二个项目来过滤这个列表，这很容易：filter（lambda项目：项目[1]




Answer 1

我用这个解决了它：

from typing import Dict, List, Any, Set

d = {"level":[1,2,3], "conf":[-1,1,2], "text":["-1","hel","llo"]}

filtered_indexes = set([i for i in range(len(d.get('conf', []))) if d.get('conf')[i] > 0])

def filter_dictionary(d: Dict[str, List[Any]], filtered_indexes: Set[int]) -> Dict[str, List[Any]]:
    for key, list_values in d.items():
        d[key] = [value for i, value in enumerate(list_values) if i in filtered_indexes]
    return d

print(filter_dictionary(d, filtered_indexes))

输出：

{'level': [2, 3], 'conf': [1, 2], 'text': ['hel', 'llo']}

Answer 2

我将保留有效元素（大于0的元素）的索引：

kept_keys = [i for i in range(len(my_dict['conf'])) if my_dict['conf'][i] > 0]

然后，您可以过滤每个列表，检查列表中某个元素的索引是否包含在keep_keys中：

{k: list(map(lambda x: x[1], filter(lambda x: x[0] in kept_keys, enumerate(my_dict[k])))) for k in my_dict}

输出：

{'level': [2, 3], 'conf': [1, 2], 'text': ['hel', 'llo']}

Answer 3

尝试：

from operator import itemgetter


def filter_dictionary(d):
    conf = d['conf']
    positive_indices = [i for i, item in enumerate(conf) if item > 0]
    f = itemgetter(*positive_indices)
    return {k: list(f(v)) for k, v in d.items()}


d = {"level": [1, 2, 3], "conf": [-1, 1, 2], "text": ["-1", "hel", "llo"]}
print(filter_dictionary(d))

输出：

{'level': [2, 3], 'conf': [1, 2], 'text': ['hel', 'llo']}

我试着先看看'conf'的哪些索引是正的，然后用itemgetter从字典中的值中选择这些索引。