如何按同农场同ripening唯一规则填充苹果variety列缺失值

根据给定约束（同一农场内相同ripening指标对应唯一苹果品种），我们可以通过分组填充的方式完成缺失值补全，具体操作如下：

实现代码 #

如果你的variety列缺失值为空字符串''，先统一转为pandas可识别的缺失值，再按规则填充即可：

import pandas as pd
import numpy as np

# 1. 将空字符串替换为pandas可识别的缺失值
df['variety'] = df['variety'].replace('', pd.NA)

# 2. 按「农场+成熟度」分组，用每组内的唯一有效品种名填充所有缺失值
df['variety'] = df.groupby(['farm', 'ripening'])['variety'].transform('first')

方案说明 #

• groupby(['farm', 'ripening'])严格遵循约束的对应关系分组，确保每组内只有一个正确的variety取值
• transform('first')会提取每组第一个非缺失的品种值，批量填充组内所有空值，运算效率高，适配大尺寸DataFrame
• 方案完全符合题目给出的前提假设，对示例数据运行后即可得到你提供的期望输出结果

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