Pandas: принудительная установка позиции столбца в кадре данных (без знания всех столбцов)

Question

Допустим, у меня есть DataFrame и я не знаю имен всех столбцов. Тем не менее, я знаю, что есть столбец с именем "N_DOC", и я хочу, чтобы это был первый столбец DataFrame (при сохранении всех других столбцов, независимо от его порядка).

Как я могу это сделать?

Answer 1

Вы можете изменить порядок столбцов таблицы с помощью reindex:

cols = df.columns.tolist()
cols.remove('N_DOC')
df.reindex(['N_DOC'] + cols, axis=1)

Answer 2

Вот простое решение, состоящее из одной строки, использующее маскирование данных:

import pandas as pd

# Building sample dataset.
cols = ['N_DOCa', 'N_DOCb', 'N_DOCc', 'N_DOCd', 'N_DOCe', 'N_DOC']
df = pd.DataFrame(columns=cols)

# Re-order columns.
df = df[['N_DOC'] + df.columns.drop('N_DOC').tolist()]

До:

Index(['N_DOCa', 'N_DOCb', 'N_DOCc', 'N_DOCd', 'N_DOCe', 'N_DOC'], dtype='object')

После:

Index(['N_DOC', 'N_DOCa', 'N_DOCb', 'N_DOCc', 'N_DOCd', 'N_DOCe'], dtype='object')

Answer 3

Используйте DataFrame.insert с DataFrame.pop для столбца извлечения:

df = pd.DataFrame({
        'A':list('abcdef'),
         'B':[4,5,4,5,5,4],
         'C':[7,8,9,4,2,3],
         'N_DOC':[1,3,5,7,1,0],
         'E':[5,3,6,9,2,4],
         'F':list('aaabbb')
})

c = 'N_DOC'
df.insert(0, c, df.pop(c))

Или:

df.insert(0, 'N_DOC', df.pop('N_DOC'))

---

print (df)
   N_DOC  A  B  C  E  F
0      1  a  4  7  5  a
1      3  b  5  8  3  a
2      5  c  4  9  6  a
3      7  d  5  4  9  b
4      1  e  5  2  2  b
5      0  f  4  3  4  b