python - 检查不同列中变量的唯一值

Question

我目前有一个如下所示的数据框：

            Owner        Vehicle_Color
0          James              Red
1          Peter              Green
2          James              Blue
3          Sally              Blue
4          Steven             Red
5          James              Blue
6          James              Red
7          Peter              Blue

我正在尝试验证一个人是否Owner分配了一种或多种车辆颜色。请记住，我的数据框有超过一百万个不同的所有者条目（可能是重复的），最好的解决方案是什么？

标签： pythonpandasdataframe

Answer 1

一种方法可能是使用groupbyand nunique：

df.groupby('Owner')['Vehicle_Color'].nunique()

结果：

Owner
James     2
Peter     2
Sally     1
Steven    1
Name: Vehicle_Color, dtype: int64