Pandas分组shift操作触发SettingWithCopyWarning的解决方法

先还原下你的场景：你有个记录App用户操作的DataFrame，结构和示例数据是这样的：

你想用df['prev_action'] = df.groupby('session_id')['action'].shift()来添加前一个操作的列，结果功能是对的，但触发了SettingWithCopyWarning对吧？

这个警告本质是Pandas在提醒你：当前的df可能只是另一个DataFrame的视图（不是独立副本），它不确定你是想修改原数据还是这个视图，所以抛出警告避免潜在的错误。下面给你几个靠谱的解决办法，以及要不要忽略的建议：

方法1：提前创建DataFrame的独立副本 #

最简单直接的方式，先把df转成独立副本，再执行赋值操作，这样Pandas就不会有歧义了：

# 先复制出独立的DataFrame
df = df.copy()
# 再添加prev_action列
df['prev_action'] = df.groupby('session_id')['action'].shift()

方法2：用.loc明确指定赋值范围 #

通过.loc来显式指定要修改的行和列，避免链式赋值带来的不确定性：

df.loc[:, 'prev_action'] = df.groupby('session_id')['action'].shift()

关于是否可以忽略警告？ #

如果你非常确定当前操作是安全的（比如df本身就是全新创建的，不是从其他df切片来的，或者你就是要修改这个视图），可以临时关闭警告，但我不推荐长期这么做——因为它可能会掩盖其他真正的赋值错误。关闭警告的代码如下：

import pandas as pd
# 关闭链式赋值警告
pd.options.mode.chained_assignment = None

用完之后记得可以再打开：pd.options.mode.chained_assignment = 'warn'，保持对潜在问题的警惕。

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