Como eliminar a coluna Pandas DataFrame

Este artigo introduz diferentes métodos para excluir as colunas DataFrame em Pandas,

• método del
• método df.drop
• método df.pop

Vamos utilizar o mesmo DataFrame para demonstrar diferentes métodos neste artigo.

>>> from pandas import DataFrame
>>> df = DataFrame.from_items(
    [('Alfa', [1, 2, 3]), 
     ('Bravo', [4, 5, 6]), 
     ('Charlie', [7,8, 9])], 
    orient='index', 
    columns=['A', 'B', 'C'])
>>> df
         A  B  C
Alfa     1  2  3
Bravo    4  5  6
Charlie  7  8  9

Método para Apagar a Coluna DataFrame

>>> df
         A  B  C
Alfa     1  2  3
Bravo    4  5  6
Charlie  7  8  9
>>> del df['A']
>>> df
         B  C
Alfa     2  3
Bravo    5  6
Charlie  8  9

del df['col_name'] apaga a coluna DataFrame que tem os nomes como col_name.

A limitação desse método del é que ele só poderia apagar uma coluna de cada vez.

df.drop Método para apagar as colunas DataFrame

drop(
    self,
    labels=None,
    axis=0,
    index=None,
    columns=None,
    level=None,
    inplace=False,
    errors="raise",
)

O método drop cai, ou em outras palavras, remove/apaga os ‘rótulos’ especificados das linhas ou colunas.

Os ‘rótulos’ podem ser um único rótulo ou rótulos de índice ou de coluna para cair.

O axis especifica se os rótulos são descartados do índice/linha (0 ou index) ou coluna (1 ou column).

index, columns é a alternativa para especificar o eixo. drop(labels, axis=0) é igual a drop(index=labels), enquanto drop(labels, axis=1) é igual a drop(column=labels).

inplace especifica que o DataFrame é modificado no lugar se inplace = True, caso contrário, ele retorna o novo DataFrame com o DataFrame original não modificado.

>>> df
         A  B  C
Alfa     1  2  3
Bravo    4  5  6
Charlie  7  8  9
>>> df.drop(["B", "C"], axis=1)
         A
Alfa     1
Bravo    4
Charlie  7

## or equally
>>> df.drop(columns=["B", "C"])
         A
Alfa     1
Bravo    4
Charlie  7

## or drop the columns in place
>>> df.drop(columns=["B", "C"],inplace=True)
>>> df
         A
Alfa     1
Bravo    4
Charlie  7

df.pop Método para excluir a coluna DataFrame em Pandas

df.pop(item)

O método DataFrame pop devolve o item e o deixa cair do DataFrame.

>>> df
         A  B  C
Alfa     1  2  3
Bravo    4  5  6
Charlie  7  8  9
>>> df.pop("A")
Alfa       1
Bravo      4
Charlie    7
Name: A, dtype: int64
>>> df
         B  C
Alfa     2  3
Bravo    5  6
Charlie  8  9        

Os dados DataFrame são modificados no lugar, como mostrado acima.