как вставить столбец с определенным индексом столбца в pandas?
могу ли я вставить столбец с определенным индексом столбца в pandas?
import pandas as pd
df = pd.DataFrame({'l':['a','b','c','d'], 'v':[1,2,1,2]})
df['n'] = 0
это поставит столбец n
как последний столбец df
, но разве нет способа сказать df
поставить n
в начале?
2 ответа:
см. документы: http://pandas.pydata.org/pandas-docs/stable/dsintro.html#column-selection-addition-deletion
С помощью idx = 0 вставит в начале
df.insert(idx, col_name, value)
df = pd.DataFrame({'B': [1, 2, 3], 'C': [4, 5, 6]}) df Out: B C 0 1 4 1 2 5 2 3 6 idx = 0 new_col = [7, 8, 9] # can be a list, a Series, an array or a scalar df.insert(loc=idx, column='A', value=new_col) df Out: A B C 0 7 1 4 1 8 2 5 2 9 3 6
вы можете попробовать извлечь столбцы в виде списка, массировать это, как вы хотите, и переиндексировать свой фрейм данных:
>>> cols = df.columns.tolist() >>> cols = [cols[-1]]+cols[:-1] # or whatever change you need >>> df.reindex(columns=cols) n l v 0 0 a 1 1 0 b 2 2 0 c 1 3 0 d 2
EDIT: это можно сделать в одной строке ; однако, это выглядит немного некрасиво. Может быть, какое-то более чистое предложение может прийти...
>>> df.reindex(columns=['n']+df.columns[:-1].tolist()) n l v 0 0 a 1 1 0 b 2 2 0 c 1 3 0 d 2