como insiro uma coluna em um índice de coluna específico nos pandas?

189

Posso inserir uma coluna em um índice de coluna específico no pandas?

import pandas as pd
df = pd.DataFrame({'l':['a','b','c','d'], 'v':[1,2,1,2]})
df['n'] = 0

Isso colocará a coluna ncomo a última coluna de df, mas não há uma maneira de dizer dfpara colocar nno início?

HappyPy
fonte
Insira uma coluna no início (extremidade esquerda) de um DataFrame - mais soluções + solução generalizada para inserir qualquer sequência (não apenas um valor constante).
cs95 5/02/19

Respostas:

370

consulte docs: http://pandas.pydata.org/pandas-docs/stable/generated/pandas.DataFrame.insert.html

usando loc = 0 será inserido no início

df.insert(loc, column, 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
Jeff
fonte
18
Para futuros usuários, os novos parâmetros são "loc", "column" e "value" . Fonte
Peter Maguire
11

Você pode tentar extrair colunas como lista, massagear isso como quiser e reindexar seu quadro de dados:

>>> 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: isso pode ser feito em uma linha; no entanto, isso parece um pouco feio. Talvez alguma proposta mais limpa possa vir ...

>>> 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
Nic
fonte
9

Se você deseja um único valor para todas as linhas:

df.insert(0,'name_of_column','')
df['name_of_column'] = value

Editar:

Você também pode:

df.insert(0,'name_of_column',value)
Hugo Vares
fonte
0

Aqui está uma resposta muito simples para isso (apenas uma linha).

Você pode fazer isso depois de adicionar a coluna 'n' ao seu df da seguinte maneira.

import pandas as pd
df = pd.DataFrame({'l':['a','b','c','d'], 'v':[1,2,1,2]})
df['n'] = 0

df
    l   v   n
0   a   1   0
1   b   2   0
2   c   1   0
3   d   2   0

# here you can add the below code and it should work.
df = df[list('nlv')]
df

    n   l   v
0   0   a   1
1   0   b   2
2   0   c   1
3   0   d   2



However, if you have words in your columns names instead of letters. It should include two brackets around your column names. 

import pandas as pd
df = pd.DataFrame({'Upper':['a','b','c','d'], 'Lower':[1,2,1,2]})
df['Net'] = 0
df['Mid'] = 2
df['Zsore'] = 2

df

    Upper   Lower   Net Mid Zsore
0   a       1       0   2   2
1   b       2       0   2   2
2   c       1       0   2   2
3   d       2       0   2   2

# here you can add below line and it should work 
df = df[list(('Mid','Upper', 'Lower', 'Net','Zsore'))]
df

   Mid  Upper   Lower   Net Zsore
0   2   a       1       0   2
1   2   b       2       0   2
2   2   c       1       0   2
3   2   d       2       0   2
rra
fonte