Cómo obtener nombres de columnas en el marco de datos de Pandas

Mientras analizamos los conjuntos de datos reales, que a menudo tienen un tamaño muy grande, es posible que necesitemos obtener los nombres de las columnas para realizar ciertas operaciones. Analicemos cómo obtener nombres de columnas en el marco de datos de Pandas . Primero, creemos un marco de datos simple con el archivo nba.csv. 

# 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 

Ahora intentemos obtener el nombre de las columnas del conjunto de datos anterior.

Método #1: Simplemente iterando sobre columnas 

# 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)

Producción:

 Método n. ° 2: usar el atributo de columnas con el objeto de marco de datos 

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

Producción:

Método n.º 3: Uso de la función de teclas(): también proporcionará las columnas del marco de datos.

# Import pandas package 
import pandas as pd 
    
# making data frame 
data = pd.read_csv("https://media.geeksforgeeks.org/wp-content/uploads/nba.csv") 
    
# calling keys() function
print(data.keys())

Producción:

 

Método #4:  el método column.values ​​devuelve una array de índice. 

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

Producción: 

 Método n. ° 5: usar el método tolist() con valores con la lista de columnas dada. 

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

Producción:

 Método #6: Uso del método sorted() : el método sorted() devolverá la lista de columnas ordenadas en orden alfabético. 

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

Producción: