在pandas中，如何使用一列中的唯一值创建列，然后根据另一列中的值填充它？

import pandas as pd

# 创建示例数据
data = {'A': ['a', 'a', 'b', 'b', 'c', 'c'],
        'B': [1, 2, 3, 4, 5, 6]}
df = pd.DataFrame(data)

# 使用groupby和transform创建新列
df['C'] = df.groupby('A')['B'].transform(lambda x: x.unique())

# 根据另一列的值填充新列
df['C'] = df['C'].fillna(df['B'].apply(lambda x: x*2))

print(df)

   A  B    C
0  a  1  1.0
1  a  2  1.0
2  b  3  3.0
3  b  4  3.0
4  c  5  5.0
5  c  6  5.0