Pandas DataFrame的列转为List
在本文中,我们将介绍如何将Pandas DataFrame的列转为List。Pandas 是Python中很常用的数据分析库之一,它提供了DataFrame(数据帧)这个数据结构,可用于处理结构化数据。DataFrame像表格一样,由行和列组成,每一行是一个数据记录,每一列是数据记录中的一种属性。有时,我们需要将DataFrame的列转为List来方便后续的处理。
阅读更多:Pandas 教程
例子
首先,我们需要导入Pandas库:
import pandas as pd
接下来,我们使用Pandas中的read_csv()函数,读取一个CSV文件,并将其转为DataFrame对象。我们将使用这个DataFrame来演示如何将其中的列转为List。
df = pd.read_csv("sample.csv")
读取CSV文件后,我们可以查看DataFrame中的数据信息。例如,我们可以查看DataFrame的头5行数据:
print(df.head())
这应该会输出如下内容:
Name Age Gender
0 Sophia 25 Female
1 John 35 Male
2 Jane 30 Female
3 David 27 Male
4 Lily 20 Female
列转为List
默认情况下,DataFrame可以通过列名或数字对列进行索引。如果需要将DataFrame中的列转为List,我们可以使用Pandas中的tolist()函数。下面是将”Age”列转为List的示例代码:
age_list = df['Age'].tolist()
print(age_list)
这会输出以下内容:
[25, 35, 30, 27, 20]
与此类似的代码,我们可以将DataFrame中的其他列转为List。例如,将”Name”列转为List:
name_list = df['Name'].tolist()
print(name_list)
这会输出以下内容:
['Sophia', 'John', 'Jane', 'David', 'Lily']
没有列名
如果DataFrame中没有列标签,我们可以使用iloc[]函数来引用列。其中,iloc[n]表示DataFrame的第n列。例如,以下示例代码展示了如何将第0列转为List:
_first_column_list = df.iloc[:, 0].tolist()
print(_first_column_list)
这会输出以下内容:
['Sophia', 'John', 'Jane', 'David', 'Lily']
请注意,中括号中的“:”表示所有行。如果只想转换一部分数据,则”0″或”:”应更改为行的起始和结束索引。
总结
在本文中,我们介绍了如何使用Pandas将DataFrame的列转为List。只需使用Pandas提供的tolist()函数即可将DataFrame中的列转为List。
age_list = df['Age'].tolist()
name_list = df['Name'].tolist()
print(age_list)
print(name_list)
极客教程