从python中的列表字典中删除公共元素

时间:2012-12-26 15:49:27

标签: python list dictionary

我有一个列表字典,列表中包含如下字典:

my_dict = {
'list1': [{'catch': 100, 'id': '1'}, {'catch': 101, 'id': '2'}, 
          {'catch': 50, 'id': '1'}], 
'list2': [{'catch': 189, 'id': '1'}, {'catch': 120, 'id': '12'}], 
'list3': [{'catch': 140, 'id': '1'}, {'catch': 10, 'id': '100'}]
}

使用commin'id'值删除列表项并将它们存储在单独的列表中的最 pythonic 方法是什么?所以输出将是这样的:

my_dict = {
'list1': [{'catch': 101, 'id': '2'}], 
'list2': [{'catch': 120, 'id': '12'}], 
'list3': [ {'catch': 10, 'id': '100'}],
'list4': [{'catch': 100, 'id': '1'}, , {'catch': 50, 'id': '1'}, 
          {'catch': 189, 'id': '1'}, {'catch': 140, 'id': '1'}]
}

在我的程序中,我有7个与此类似的列表,如果'id'出现在这些列表中的两个或更多列表中,我想在第8个列表中存储具有'id'的项目的所有外观以供进一步处理

关心, finnurtorfa

3 个答案:

答案 0 :(得分:3)

考虑将您的数据重组为以下内容:

>>> import itertools
>>> { k: [d['catch'] for d in v] for k, v in itertools.groupby(sorted(itertools.chain(*my_dict.itervalues()), key=lambda d: d['id']), lambda d: d['id']) }
{'1': [100, 50, 140, 189], '2': [101], '100': [10], '12': [120]}

您尚未描述数据代表的内容,因此这可能不适合您。但是使用的工具(来自chain的{​​{1}}和groupby)至少应该给你一些想法。

编辑:我在我的测试中偶然使用了问题的示例答案。通过向itertools的输入添加排序来修复。

答案 1 :(得分:2)

>>> get_id = operator.itemgetter("id")
>>> flattened_dict = itertools.chain.from_iterable(my_dict.values())
>>> groups = itertools.groupby(sorted(flattened_dict, key=get_id), get_id)
>>> {k: list(v) for k, v in groups}
{'1': [{'catch': 100, 'id': '1'},
  {'catch': 50, 'id': '1'},
  {'catch': 140, 'id': '1'},
  {'catch': 189, 'id': '1'}],
 '100': [{'catch': 10, 'id': '100'}],
 '12': [{'catch': 120, 'id': '12'}],
 '2': [{'catch': 101, 'id': '2'}]}

说明:

  • get_id是一个接受对象x并返回x["id"]的函数。
  • flattened_dict只是对所有列表的可迭代(即连接.values()的所有my_dict
  • 现在我们使用键函数flattened_dictget_id进行排序 - 即按ID排序 - 并按结果对结果进行分组。

这基本上有效,因为itertools.groupby非常棒。

答案 2 :(得分:0)

以下内容:

my_dict = {
'list1': [{'catch': 100, 'id': '1'}, {'catch': 101, 'id': '2'}, 
      {'catch': 50, 'id': '1'}], 
'list2': [{'catch': 189, 'id': '1'}, {'catch': 120, 'id': '12'}], 
'list3': [{'catch': 140, 'id': '1'}, {'catch': 10, 'id': '100'}]
}

from itertools import groupby

sub = {}
for k in my_dict:
 for kk, g in groupby( my_dict[k], lambda v: v["id"] ):
   if not kk in sub:
    sub[kk] = []
   sub[kk] = sub[kk] + list( g )

print sub

{'1': [{'catch': 100, 'id': '1'}, {'catch': 50, 'id': '1'}, {'catch': 140, 'id': '1'}, {'catch': 189, 'id': '1'}], '12': [{'catch': 120, 'id': '12'}], '100': [{'catch': 10, 'id': '100'}], '2': [{'catch': 101, 'id': '2'}]}