количество уникальных значений для каждого столбца на основе специфика c значения другого столбца python

Question

Чтобы напечатать количество уникальных значений для каждого столбца в кадре данных, который я использовал:

for col in dataset:
        print(dataset[col].value_counts(), '\n')

Но я хочу добавить к нему какое-то условие, например, напечатать это число уникальных значений, если другое столбец (например, столбец «num», который имеет только два значения: 0 и 1) значение равно 0. Также я попробовал это, но не дал мне правильный ответ:

df = dataset.groupby('num')['age'].nunique()

Спасибо!

Answer 1

Вы делаете что-то вроде этого -

dataset[dataset['class'] == 0].groupby(['num'])['age'].nunique()