基于其他列的非NA值折叠Pandas列的值

import pandas as pd

# 创建一个示例DataFrame
data = {'A': [1, 2, 3, 4, 5],
        'B': [10, None, 30, 40, None],
        'C': [100, 200, None, None, 500]}
df = pd.DataFrame(data)

# 基于列A的非NA值折叠列B的值
df['B'] = df.groupby('A')['B'].transform(lambda x: x.dropna().iloc[0])

# 基于列A的非NA值折叠列C的值
df['C'] = df.groupby('A')['C'].transform(lambda x: x.dropna().iloc[0])

print(df)

   A     B      C
0  1  10.0  100.0
1  2   NaN  200.0
2  3  30.0    NaN
3  4  40.0    NaN
4  5   NaN  500.0