Как удалить несколько (~ 5000) столбцов в кадре данных панд?

Question

У меня есть датафрейм с 5632 столбцами, и я хочу сохранить только 500 из них.У меня есть имена столбцов (которые я хочу сохранить) в кадре данных, с именами в качестве индекса строки.Есть ли способ сделать это?

Answer 1

Предположим, ваш DataFrame назван df, и у вас есть список cols индексов столбцов, которые вы хотите сохранить.Затем вы должны использовать:

df1 = df.iloc[:, cols]

Этот оператор отбрасывает все столбцы, кроме тех, чьи индексы были указаны в cols.Используйте df1 в качестве нового DataFrame.