在同一组中用相同的代码字符填写 NA 值

Question

我有这个数据集。 我想在同一组中用相同的代码字符填写 NA 值。

在此示例中，第一个 NA 将是 S，第二个将是 F

谢谢，

df = {'Key': ['111*1', '111*2','111*3', '222*1','222*2', '333*1','333*2', '333*3','333*4', '444*1'],
        'code': ['S', 'S','NA', 'M','M', 'F','F', 'F','NA', 'C']}
  
# Create DataFrame
df = pd.DataFrame(df)
df[['Keya', 'Keyb']] = df['Key'].str.split('\\*', expand=True, regex=True)
print(df)

Answer 1

您可以使用：

s = df['code'].replace('NA', np.nan)

df['code'] = s.fillna(s.groupby(df['Key'].str.extract('([^*]+)*', expand=False))
                       .transform('first')
                     )

如果您已经拥有 Keya：

s = df['code'].replace('NA', np.nan)

df['code'] = s.fillna(s.groupby(df['Keya']).transform('first'))

Output：

     Key code
0  111*1    S
1  111*2    S
2  111*3    S
3  222*1    M
4  222*2    M
5  333*1    F
6  333*2    F
7  333*3    F
8  333*4    F
9  444*1    C