Проверка уникального значения для переменной в другом столбце

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

И я пытаюсь проверить, назначен ли один владелец одному или нескольким цветам автомобиля.Учитывая, что в моем фрейме данных содержится более миллиона записей для владельцев (которые могут быть повторены), какое решение будет лучшим?

Спасибо

Answer 1

Одним из способов может быть использование groupby и nunique:

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

Результаты:

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