Как вывести список столбцов из фрейма данных Pandas на основе уникального количества значений?

Question

Я новичок в python и пытаюсь отфильтровать столбцы из кадра данных на основе уникального числа значений.Я хочу создать список столбцов, число уникальных значений которых равно 3. Ниже мой код:

cols=objects_ds.apply(pd.Series.nunique) ==3

Это дает мне блок данных с логическими значениями (True / False), как показано ниже:

A FALSE
B FALSE
C TRUE
D TRUE
E TRUE
F FALSE

Я хочу, чтобы мои выходные данные были следующими:

cols=['C','D','E']

Но я не уверен, как извлечь имена столбцов на основе логических значений.Пожалуйста, помогите!

Спасибо

Answer 1

Предполагая, что df является вашим фреймом данных:

ser = df.nunique() == 3
ser[ser.values].index

Answer 2

Вам не нужно подавать заявку здесь

df.columns[(df.nunique()==3)].tolist()