python - 在 pandas 中创建质量分数列

Question

您好，我正在使用 pandas 中的数据框，它看起来像这样

ID    Color    Size      Shape
1     Blue     Small     Triangle
2     Red      Medium    Square
3     Yellow   Large     Circle

我想将每一行与数据列表进行比较，并创建一个新的分数列来计算每行与列表匹配的次数。

示例 [Red、Medium、Circle] 将生成以下数据帧。

ID    Color    Size      Shape      Score
1     Blue     Small     Triangle     0
2     Red      Medium    Square       2
3     Yellow   Large     Circle       1

理想情况下，我希望能够创建多个分数列来检查多个列表。

Answer 1

对 data.frame 使用 isin

l=['Red', 'Medium', 'Circle']
df['score']=df.isin(l).sum(1)
df
Out[404]: 
   ID   Color    Size     Shape  score
0   1    Blue   Small  Triangle      0
1   2     Red  Medium    Square      2
2   3  Yellow   Large    Circle      1