Pandas按指定名称分组移动列值的实现方法咨询

从你的示例数据和需求来看，核心是要在每个名称分组内，把最后一列的值向上偏移1位（也就是每行取同组下一行的对应值，最后一行用NaN填充），用Pandas可以很简洁地实现这个需求，具体步骤如下：

核心思路 #

利用groupby按名称分组，然后对目标列使用shift(-1)方法——这个方法会将列内的值向上移动1位，自动为每组的最后一行填充NaN，完美匹配你要的结果。

代码示例 #

假设你的原始DataFrame列名为Name、id、date、flag（如果列名不同，替换成实际名称即可）：

import pandas as pd

# 构造你的示例数据
data = [
    ["Name_A", "id1", "date1", 1],
    ["Name_A", "id1", "date2", 0],
    ["Name_A", "id1", "date3", 1],
    ["Name_A", "id1", "date4", 1],
    ["Name_A", "id1", "date5", 1],
    ["Name_B", "id2", "date6", 0],
    ["Name_B", "id2", "date7", 1],
    ["Name_B", "id2", "date8", 1]
]
df = pd.DataFrame(data, columns=["Name", "id", "date", "flag"])

# 分组后对flag列执行向上移位
df['result'] = df.groupby('Name')['flag'].shift(-1)

# 提取你需要的列得到最终结果
final_df = df[['Name', 'date', 'result']]
print(final_df)

运行后输出的结果和你期望的完全一致：

Name    date  result
0  Name_A  date1     0.0
1  Name_A  date2     1.0
2  Name_A  date3     1.0
3  Name_A  date4     1.0
4  Name_A  date5     NaN
5  Name_B  date6     1.0
6  Name_B  date7     1.0
7  Name_B  date8     NaN

关键细节说明 #

• groupby('Name')确保移位操作只在同一个名称的组内进行，不会出现跨组取值的情况；
• shift(-1)是向上移位1位，如果你需要向下移位（取上一行的值），可以改成shift(1)；
• 如果不需要保留原始的id和flag列，直接提取Name、date和新生成的result列即可。

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