如何在Pandas数据框架中获得列名

在分析真实的数据集时，这些数据集的大小往往是非常巨大的，我们可能需要获得列名，以执行一些特定的操作。让我们来讨论一下如何在Pandas数据框架中获取列名。首先，让我们用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 

现在，让我们尝试从上述数据集中获取列名。

方法#1：简单地在列上进行迭代

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

输出:

方法#2：使用数据框架对象的列属性

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

输出:

方法#3：使用keys()函数。它也会给出数据框架的列。

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

输出:

方法#4： column.values方法返回一个数组的索引。

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

输出:

方法#5：使用tolist()方法，用给定的列的值来计算。

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

输出:

方法#6：使用sorted()方法：sorted()方法将返回按字母顺序排序的列列表。

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

输出: