如何在for循环中使用Pandas groupby并消除FutureWarning警告

方法一：直接使用列名字符串代替单元素列表 #

把分组键从列表['col1']改为字符串'col1'，迭代时i直接对应分组的col1值，既消除警告，还能简化代码：

for i, g in df2.groupby('col1'):
    col1_val = i  # 无需再从分组DataFrame中提取，i就是分组的col1值
    print(col1_val)

方法二：解构分组键元组（兼容动态分组场景） #

如果实际代码中分组键是动态生成的（可能是单元素或多元素列表），可以在循环时显式解构长度为1的元组，适配未来版本的返回格式：

for (i,), g in df2.groupby(['col1']):
    col1_val = i
    print(col1_val)

警告原因说明 #

Pandas的警告提示：未来版本中，当分组键是长度为1的列表时，迭代groupby对象会返回长度为1的元组（比如(0,)而非0）。上述两种方法分别从避免触发该行为、适配未来返回格式两个角度解决问题。

内容的提问来源于stack exchange，提问作者edn