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

R语言中按月份拆分DataFrame并将循环输出保存为列表求助

按month_key拆分DataFrame并保存为列表的实现方案

嘿,这事儿其实挺容易搞定的,我给你分享两种常用的方法,都是pandas里的常规操作:

方法1:用groupby快速生成列表

这是最简洁高效的方式,pandas的groupby工具可以直接帮你完成分组,再转成列表就行:

import pandas as pd

# 假设你的原始DataFrame名为df
split_dfs = [group for _, group in df.groupby('month_key')]

简单解释下:df.groupby('month_key')会把原数据按month_key的每个唯一值分成对应的组,之后用列表推导式把每个分组对应的小DataFrame提取出来——这里的_是个占位符,用来接收每个分组的键值(也就是具体的month_key数值),咱们不需要用到这个键,所以用下划线代替就行。这样得到的split_dfs就是包含144个小DataFrame的列表啦。

方法2:手动循环筛选(更直观)

如果你想一步步看到拆分过程,或者需要在拆分时加一些额外操作,可以先获取month_key的所有唯一值,再循环筛选:

# 先拿到所有不重复的month_key值
unique_month_keys = df['month_key'].unique()

# 初始化一个空列表用来存拆分后的小DataFrame
split_dfs = []

# 循环每个唯一的month_key,筛选出对应的数据并加入列表
for key in unique_month_keys:
    # 这里用.copy()是为了避免得到原DataFrame的切片视图,确保每个小DF都是独立的
    small_df = df[df['month_key'] == key].copy()
    split_dfs.append(small_df)

验证结果的小技巧

拆分完之后可以简单验证下是否符合预期:

  • 检查列表长度是否为144:print(len(split_dfs))
  • 查看某个小DataFrame的month_key是否唯一:print(split_dfs[0]['month_key'].unique())

内容的提问来源于stack exchange,提问作者Prasanna Tatiparthi

相关产品推荐
方舟 Agent Plan

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

最近更新时间:2026.05.21 06:25:17