python pandas DataFrame - 将列表分配给多个单元格

Question

我有一个像这样的数据框

name col1 col2
 a    aa   123
 a    bb   123
 b    aa   234

和一个清单

[1, 2, 3]

我想用 col1 = 'aa' 替换每行的 col2 ，列表如下

name col1     col2
 a    aa   [1, 2, 3]
 a    bb       123
 b    aa   [1, 2, 3]

我尝试过类似的东西

df.loc[df[col1] == 'aa', col2] = [1, 2, 3]

但它给了我错误：

name col1 col2
 a    aa   123
 a    bb   123
 b    aa   234

我应该如何解决这个问题？

Answer 1

import pandas as pd
df = pd.DataFrame({"name":["a","a","b"],"col1":["aa","bb","aa"],"col2":[123,123,234]})
l = [1,2,3]
df["col2"] = df.apply(lambda x: l if x.col1 == "aa" else x.col2, axis =1)
df