python - 如何从基于三列的DataFrame中提取数据？

Question

我试图从基于三列的数据框中洗牌并提取重复的列。

我尝试将三列转换为字典并存储它们的索引并比较高度。第 4 行不是唯一删除的。

df['C']=df[["Color1","Color2","Color3"]].stack().apply(tuple)
df = df.duplicated(subset=["Color1","Color2","Color3"], keep=False)


     Height    Color1    Color2    Color3
0    Short      NaN       Blue      Red
1    High       Red       Blue      NaN
2    Medium     Blue       Red      NaN 
3    Short       NaN       NaN      Blue
4    Short       NaN       Red      Blue
5    High        NaN       NaN      NaN

代码的输出应该是:

     Height    Color1    Color2    Color3
0    Short      NaN       Blue      Red
1    High       Red       Blue      NaN
2    Medium     Blue       Red      NaN

Answer 1

您可以使用drop_duplicates :

df.drop_duplicates(subset="Height")