如何在Pandas DataFrame中仅对特定行子集进行排序？

原始DataFrame #

需求说明 #

仅对is_plane == False的连续行块进行排序，is_plane == True的行必须保持原始位置不变，最终得到目标DataFrame。

解决方案 #

利用pandas的cumsum生成分组键，结合groupby对不同行块进行针对性处理：

import pandas as pd

# 构造原始DataFrame
data = {
    'Plane Parts': ['G6_32 FAB', 'G6_32 KIT', 'Item D', 'Item C', 'Item A', 
                    'G6_32 SITE', 'G6_32 SPACE', 'Item C', 'Item A', 'Item F'],
    'Quantity': [1, 2, 2, 4, 5, 5, 6, 2, 1, 2],
    'is_plane': [True, True, False, False, False, True, True, False, False, False]
}
df = pd.DataFrame(data)

# 生成分组键：用True行作为分隔，将连续的False行划分为独立组
df['group_key'] = df['is_plane'].cumsum()

# 分组处理：False组按Plane Parts升序排序，True组直接保留原顺序
result = df.groupby('group_key', group_keys=False).apply(
    lambda x: x.sort_values('Plane Parts') if not x['is_plane'].any() else x
)

# 清理临时列并重置索引
result = result.drop(columns='group_key').reset_index(drop=True)

# 输出结果
print(result)

结果验证 #

运行后得到的结果与目标一致：

原理说明 #

• df['is_plane'].cumsum()：由于True在数值运算中等同于1，False等同于0，累加后会为每个连续的False行块分配唯一的分组键，True行也会各自成组。
• groupby分组后，针对每组判断是否包含True行：不包含的（纯False组）执行排序，包含的（True组）直接保留原始顺序，从而实现需求。

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