获取Pandas数据框架的指定列的列表
在这篇文章中,我们将讨论如何获得一个Pandas数据框架的指定列的列表。首先,我们将把一个csv文件读入一个pandas数据框架。
注意:要获得所用的CSV文件,请点击这里。
示例:
# importing pandas module
import pandas as pd
# making data frame from csv
data = pd.read_csv("nba.csv")
# calling head() method
df = data.head(5)
# displaying data
df
输出:
让我们看看如何获得Pandas数据框架的指定列的列表:
我们将用三种不同的方式将 “姓名 “列转换为列表。
1.使用 系列.tolist()
从数据框架中,我们使用[]操作符选择列 “Name”,该操作符返回一个系列对象。接下来,我们将使用系列类提供的函数Series.to_list()来转换系列对象并返回一个列表。
# importing pandas module
import pandas as pd
# making data frame from csv
data = pd.read_csv("nba.csv")
df = data.head(5)
# Converting a specific Dataframe
# column to list using Series.tolist()
Name_list = df["Name"].tolist()
print("Converting name to list:")
# displaying list
Name_list
输出:
让我们把它分解开来,看看有哪些类型
# column 'Name' as series object
print(type(df["Name"]))
# Convert series object to a list
print(type(df["Name"].values.tolist()
输出:
2.使用numpy.ndarray.tolist()
我们从数据框架中选择 “Name “列,使用[]操作符返回一个系列对象,并使用Series.Values从系列对象中获得一个NumPy数组。接下来,我们将使用NumPy数组提供的函数tolist()将其转换为一个列表。
# importing pandas module
import pandas as pd
# making data frame from csv
data = pd.read_csv("nba.csv")
df = data.head(5)
# Converting a specific Dataframe column
# to list using numpy.ndarray.tolist()
Name_list = df["Name"].values.tolist()
print("Converting name to list:")
# displaying list
Name_list
输出:
同样地,将其分解
# Select a column from dataframe
# as series and get a numpy array
print(type(df["Name"].values))
# Convert numpy array to a list
print(type(df["Name"].values.tolist()
输出:
3.使用Python list()函数
你也可以使用Python的list()函数,带一个可选的iterable参数,将一个列转换成一个列表。
# importing pandas module
import pandas as pd
# making data frame from csv
data = pd.read_csv("nba.csv")
df = data.head(5)
# Converting a specific Dataframe
# column to list using list()
# function in Python
Name_List = list(df["Name"])
print("Converting name to list:")
# displaying list
Name_List
输出:
将索引列转换为列表。
索引列可以转换为列表,方法是调用pandas.DataFrame.index,它将索引列作为一个数组返回,然后调用index_column.tolist(),将index_column转换为一个列表。
# Converting index column to list
index_list = df.index.tolist()
print("Converting index to list:")
# display index as list
index_list
输出: