Ao analisar os conjuntos de dados reais, que geralmente são muito grandes, podemos precisar obter os nomes das colunas para realizar algumas operações.

Vamos discutir como obter nomes de colunas no dataframe do Pandas .

Primeiro, vamos criar um dataframe simples com nba.csvarquivo.

# Import pandas package 
import pandas as pd 
    
# making data frame 
data = pd.read_csv("https://media.geeksforgeeks.org/wp-content/uploads/nba.csv") 
    
# calling head() method  
# storing in new variable 
data_top = data.head() 
    
# display 
data_top 

Agora vamos tentar obter o nome das colunas do conjunto de dados acima.

<iframe allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen="" frameborder="0" height="374" src="https://www.youtube.com/embed/Ty9TWZ7js6A?feature=oembed" width="665"></iframe>

Método # 1: simplesmente iterar nas colunas

# Import pandas package 
import pandas as pd 
    
# making data frame 
data = pd.read_csv("nba.csv") 
  
# iterating the columns
for col in data.columns:
    print(col)

Saída:

 
Método 2: Usando colunas com objeto dataframe

# Import pandas package 
import pandas as pd 
    
# making data frame 
data = pd.read_csv("nba.csv") 
    
# list(data) or
list(data.columns)

Saída:

 
Método # 3: o column.values método retorna uma matriz de índice.

# Import pandas package 
import pandas as pd 
    
# making data frame 
data = pd.read_csv("nba.csv") 
    
list(data.columns.values)

Saída:

 
Método 4: Usando o tolist()método com valores com a lista de colunas fornecida.

# Import pandas package 
import pandas as pd 
    
# making data frame 
data = pd.read_csv("nba.csv") 
    
list(data.columns.values.tolist())

Resultado:

 
Método nº 5:sorted() Método de uso

O método Sorted() retornará a lista de colunas classificadas em ordem alfabética.

# Import pandas package 
import pandas as pd 
    
# making data frame 
data = pd.read_csv("nba.csv") 
    
# using sorted() method
sorted(data)

Saída: