假设我们有如下的嵌套列表结构:
nested_list = [
{'name': 'foo', 'value': 1},
{'name': 'bar', 'value': 2},
{'name': 'foo', 'value': 3},
{'name': 'baz', 'value': 4},
{'name': 'bar', 'value': 5},
]
我们希望将所有 'foo'、'bar'、'baz' 元素分别放在一起,构建一个新的嵌套列表。
我们可以使用 defaultdict(list) 将具有相同名称的元素放在同一个列表中,然后使用字典的 values() 方法获取字典中所有的值。最后将所有的值合并成一个列表即可。
完整的代码示例如下所示:
from collections import defaultdict
nested_list = [
{'name': 'foo', 'value': 1},
{'name': 'bar', 'value': 2},
{'name': 'foo', 'value': 3},
{'name': 'baz', 'value': 4},
{'name': 'bar', 'value': 5},
]
# 将具有相同名称的元素放在同一个列表中
d = defaultdict(list)
for item in nested_list:
d[item['name']].append(item)
# 获取字典中所有的值
result = list(d.values())
# 合并所有的值为一个列表
new_nested_list = [item for sublist in result for item in sublist]
print(new_nested_list)
以上代码将输出以下结果:
[
{'name': 'foo', 'value': 1},
{'name': 'foo', 'value': 3},
{'name': 'bar', 'value': 2},
{'name': 'bar', 'value': 5},
{'name': 'baz', 'value': 4}
]
上一篇:按元素名称过滤表格