Pandas数据帧迭代对比Site修改EndTime遇警告求助

直接用Pandas的移位（shift）和布尔索引实现，既避免警告又保证效率：

1. 生成移位后的Site列，获取下一行和下下一行的Site值
2. 构建三者均不同的布尔条件
3. 用loc安全地更新符合条件的行的EndTime值

示例代码：

import pandas as pd

# 构造示例DataFrame
data = {
    'Site': ['A', 'B', 'C', 'A', 'D', 'E'],
    'EventTime': ['2023-01-01', '2023-01-02', '2023-01-03', '2023-01-04', '2023-01-05', '2023-01-06'],
    'EndTime': ['2023-01-01 10:00', '2023-01-02 10:00', '2023-01-03 10:00', '2023-01-04 10:00', '2023-01-05 10:00', '2023-01-06 10:00']
}
df = pd.DataFrame(data)

# 获取下一行和下下一行的Site
next_site = df['Site'].shift(-1)
next_next_site = df['Site'].shift(-2)

# 构建三者均不同的条件
condition = (df['Site'] != next_site) & (df['Site'] != next_next_site) & (next_site != next_next_site)

# 用loc更新符合条件的EndTime
df.loc[condition, 'EndTime'] = df['EndTime'].shift(-2)

print(df)

为什么你的循环会出问题？ #

• SettingWithCopyWarning：循环中直接修改DataFrame的行时，可能是在操作原DataFrame的视图而非副本，Pandas会抛出警告提示你可能意外修改了原始数据。
• 结果不符合预期：循环处理时容易忽略边界情况（比如最后两行没有下下一行），或者索引处理错误，导致赋值逻辑出错。

关键说明 #

• shift(-1)：将列向下移动一位，获取下一行的值；shift(-2)获取下下一行的值
• loc：Pandas中推荐的安全赋值方式，确保你是在修改原DataFrame的有效数据，避免警告
• 边界行（最后两行）：shift(-2)会返回NaN，对应的条件会自动为False，所以不会被处理，符合需求

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