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

#python #list #multiple-columns

#питон #Список #несколько столбцов

Вопрос:

У меня есть набор данных из 45 столбцов, и я хотел бы извлечь только числовые типы (int64 и float64).

Я попробовал это

 mylist = list(df.select_dtypes(include=[np.number]).columns)   

… дал мне список из 28 имен

Теперь я мог бы сделать

 num_df = df[['col1','col13']]  

но это было бы слишком долго …

Ответ №1:

что-то вроде этого :

 df.select_dtypes(include=['int', 'float'])