You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

复杂列表元素筛选:按首元素分组拆分嵌套结构列表

解决列表按首元素分组的问题

嘿,我来帮你搞定这个分组需求~首先得说说你之前用的代码为啥行不通:你写的df1.items()是字典专属的方法,但df1其实是列表而非字典,直接调用会抛出AttributeError,而且逻辑上也没实现分组的意图。

正确实现方法

我们可以通过「字典分组」的思路来完成需求,把相同首元素的元组归类到一起,再提取对应的分组即可。这里有两种实现方式:

方式一:基础字典实现

df1 = [('f', {'abe': 1}), ('f', {'tbeli': 1}), ('g', {'mos': 1}), ('g', {'esc': 1})]

# 初始化分组字典
grouped_dict = {}
for key, val in df1:
    # 如果当前key不在字典中,先创建对应的空列表
    if key not in grouped_dict:
        grouped_dict[key] = []
    # 将原元组添加到对应分组的列表中
    grouped_dict[key].append((key, val))

# 提取目标分组
df2 = grouped_dict['f']
df3 = grouped_dict['g']

print(df2)  # 输出: [('f', {'abe': 1}), ('f', {'tbeli': 1})]
print(df3)  # 输出: [('g', {'mos': 1}), ('g', {'esc': 1})]

方式二:用collections.defaultdict简化代码

如果想让代码更简洁,可以用Python标准库的defaultdict,它会自动为不存在的key创建默认值(这里是空列表):

from collections import defaultdict

df1 = [('f', {'abe': 1}), ('f', {'tbeli': 1}), ('g', {'mos': 1}), ('g', {'esc': 1})]

grouped_dict = defaultdict(list)
for key, val in df1:
    grouped_dict[key].append((key, val))

df2 = grouped_dict['f']
df3 = grouped_dict['g']

核心逻辑说明

两种方式的核心都是遍历原列表df1,把每个元组按第一个元素('f'或'g')作为key,归类到字典对应的列表中,最后直接从字典中取出对应key的列表,就是你需要的df2和df3啦~

内容的提问来源于stack exchange,提问作者just3 just5

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.05.25 06:13:48