Как удалить несколько столбцов в кадре данных, используя одновременно с loc, а также с именами столбцов - PullRequest
2 голосов
/ 16 апреля 2019

Я хочу отбросить несколько столбцов из фрейма данных, из которых несколько столбцов расположены непрерывно, и их можно сгруппировать с помощью df.loc.

Но у меня также есть несколько имен столбцов, которые нужно отбрасывать, используя сами имена столбцов.

Может кто-то мне помочь, используя имена df.loc и столбцов одновременно в одном и том же наброске.

df_modified = df.drop(df.loc[:, 'Quarter of Joining': 'Age in Company (Years)'].columns, axis =1)

Я также хочу добавить еще несколько столбцов с их именем, например, 'DOB', 'Gender'

1 Ответ

3 голосов
/ 16 апреля 2019

На мой взгляд, ваше решение проще, вот альтернатива:

another = ['DOB','Gender']
m1 = (df.columns.values == 'Quarter of Joining').cumsum() != 0
m2 = (df.columns.values[::-1] == 'Age in Company (Years)').cumsum() != 0

df_modified = df = df.drop(df.columns[m1 & m2[::-1]].union(another), axis=1)

Ваше решение должно быть изменено:

another = ['DOB','Gender']
c = df.loc[:, 'Quarter of Joining': 'Age in Company (Years)'].columns

df_modified = df.drop(c.union(another), axis=1)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...