Извлеките список категориальных имен на основе уникальных уровней

#python #pandas #dataframe

Вопрос:

Я отрабатываю свои навыки работы на python, решая онлайн-упражнения. Я хотел бы получить некоторую помощь в этом вопросе, потому что я никогда не слышал об уровнях в python (я знаю, что они доступны в R).

Выполните функцию, чтобы извлечь список имен категориальных объектов из фрейма данных, используя пороговое значение для вывода категориальных переменных на основе количества уникальных уровней. Исключите целевую переменную и отсортируйте полученный список.

 import pandas as pd

def extract_categorical_features(df, n_levels, target):
    
    # Write your code here...
    solution = None
    return solution

# Click 'Run' to execute test case
test_case = extract_categorical_features(churn_df, 6, 'Churn')
 

Комментарии:

1. можете ли вы показать, что вы пробовали? В панд есть некоторая документация о категориях pandas.pydata.org/pandas-docs/stable/user_guide/…

Ответ №1:

Я думаю, что это то, о чем они просят:

 cat_feats = []
[cat_feats.append(col) for col in df.columns if len(df[col].unique()) <= n_levels]
return sorted(cat_feats)