如何在 Pandas 数据框中获取列名

在分析通常非常庞大的真实数据集时，我们可能需要获取列名才能执行某些操作。让我们讨论如何在 Pandas 数据框中获取列名。首先，让我们使用 nba.csv 文件创建一个简单的数据框。

Python3

# 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：简单地遍历列

Python3

# 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：使用带有数据框对象的列

Python3

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

输出： 方法#3： column.values 方法返回一个索引数组。

Python3

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

输出： 方法 #4：使用 tolist() 方法和给定列列表的值。

Python3

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

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

Python3

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

输出：