Como listar todos os nomes de colunas em pandas (4 métodos)

Você pode usar um dos quatro métodos a seguir para listar todos os nomes de colunas em um DataFrame do pandas:

Método 1: usando ganchos

 [column for column in df]

Método 2: use tolist()

 df. columns . values . tolist ()

Método 3: use lista()

 list(df)

Método 4: use list() com valores de coluna

 list(df. columns . values )

Os exemplos a seguir mostram como usar cada um desses métodos com o seguinte DataFrame do pandas:

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' points ': [25, 12, 15, 14, 19, 23],
                   ' assists ': [5, 7, 7, 9, 12, 9],
                   ' rebounds ': [11, 8, 10, 6, 6, 5],
                   ' blocks ': [6, 6, 3, 2, 7, 9]})

#view DataFrame
df

	points assists rebounds blocks
0 25 5 11 6
1 12 7 8 6
2 15 7 10 3
3 14 9 6 2
4 19 12 6 7
5 23 9 5 9

Método 1: usando ganchos

O código a seguir mostra como listar todos os nomes de colunas em um DataFrame do pandas usando colchetes:

 [column for column in df]

['points', 'assists', 'rebounds', 'blocks']

Método 2: use tolist()

O código a seguir mostra como listar todos os nomes de colunas usando a função .tolist() :

 df. columns . values . tolist ()

['points', 'assists', 'rebounds', 'blocks'] 

Método 3: use lista()

O código a seguir mostra como listar todos os nomes de colunas usando a função list() :

 list(df)

['points', 'assists', 'rebounds', 'blocks'] 

Método 4: use list() com valores de coluna

O código a seguir mostra como listar todos os nomes de colunas usando a função list() com valores de coluna:

 list(df. columns . values )

['points', 'assists', 'rebounds', 'blocks']

Observe que todos os quatro métodos retornam os mesmos resultados.

Observe que para DataFrames extremamente grandes, o método df.columns.values.tolist() tende a ser o mais rápido.

Recursos adicionais

Os tutoriais a seguir explicam como executar outras funções comuns com as colunas de um DataFrame do pandas:

Como deletar colunas no Pandas
Como excluir colunas no Pandas
Como aplicar uma função a colunas selecionadas no Pandas
Como alterar a ordem das colunas no Pandas DataFrame