python - 用列表中的值替换 pandas 子集中的所有值

Question

我有一个由 3 个不同组 A、B、C 组成的数据集:

df = pd.DataFrame([[1, "A", "A"], [2, "B", "B"], [3, "A", "A"], [4, "B", "B"], [5, "C", "C"], [6, "C", "C"],
               [7, "A", "A"], [8, "B", "B"], [9, "C", "C"]],
              columns=["Index", "Group", "Cluster"])

对于每个组，我确定一些集群，我想用集群的值替换集群列。每个集群都有一个列表

listA = [0, 1, 1]
listB = [1, 0, 1]
listC = [0, 0, 1]

我想在最后有这样的东西:

<表类="s-表"><头>索引组集群<正文>1一个02B13一个14B05C06C07一个18B19C1

Answer 1

试试for循环

for x, y in zip(list('ABC'), [listA,listB,listC]):
     df.loc[df.Group == x, 'Cluster'] = y
df
Out[395]: 
   Index Group Cluster
0      1     A       0
1      2     B       1
2      3     A       1
3      4     B       0
4      5     C       0
5      6     C       0
6      7     A       1
7      8     B       1
8      9     C       1