列出具有大于特定数字pandas的独特值的分类列列表

Question

我有一个带有分类，数字和日期列的DF。 分类列的列表。

 date_time1      date_time2     cat_col1    cat_col_2    num_col1   num_col2   cat_col3
 2020-10-08       2021-11-08       ABC         xyz          20         40        PQR
 19:09:21.884     15:18:26.864
 2020-10-08       2021-11-08       BCD         xyz          30         50        ABC     
 19:09:21.884     15:18:26.864
 2020-10-08       2021-11-08       ABC         yza          40         30        MNO
 19:09:21.884     15:18:26.864  
 2020-10-08       2021-11-08       CDE         xyz          10         80        CDE
 19:09:21.884     15:18:26.864
 2020-10-08       2021-11-08       BCD         xyz          20         70        MNO
 19:09:21.884     15:18:26.864
  

我想列出所有具有唯一值超过2的 应该

 mylist =['cat_col1', 'cat_col3']

有人可以帮我吗？

Answer 1

如果要仅按名称选择列：

[col for col in df.columns if col.startswith('cat_') and df[col].nunique() > 2]

结果：

['cat_col1', 'cat_col3']

如果要按类型选择：

[col for col in df.select_dtypes(include='category').columns if df[col].nunique() > 2]