Блог

Доброе утро,

фреймы данных, с которыми я работаю, всегда (100%) содержат четыре определенных столбца, но остальные из них различаются! Есть ли способ сохранить только определенные столбцы и не отбрасывать их? Прямо сейчас я использую этот код для выполнения…

 df = df.drop(["column_a", "column_b", "column_c", axis=1)
 

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

Спасибо и желаю вам отличных выходных!

 df = df[['col_to_keep_1', 'col_to_keep_2', 'col_to_keep_3', 'col_to_keep_4']]
 

если вы хотите удалить все столбцы, кроме некоторых, вы можете использовать это:

 df = df.loc[:, ['a','b']]
 

Где df['a'] и df['b'] какие колонки хранить.