📌  相关文章
📜  python - 将数据框中的列转换为列表 - Python (1)

📅  最后修改于: 2023-12-03 15:33:56.951000             🧑  作者: Mango

Python - 将数据框中的列转换为列表

如果你需要将数据框中的列转换为列表,Python提供了多种方法来实现这一目的。下面我们将介绍其中的几种方法。

使用pandas

如果你在处理的数据已经是一个pandas数据框,那么可以使用pandas自带的方法来将列转换为列表。例如,假设我们有以下数据框:

import pandas as pd

data = {
    "name": ["Alice", "Bob", "Charlie"],
    "age": [25, 30, 35],
    "city": ["New York", "London", "Paris"]
}

df = pd.DataFrame(data)

我们可以将"age"列转换为列表,方法如下:

age_list = df["age"].tolist()

这将生成一个名为age_list的列表,其中包含了数据框中"age"列的所有数据。

使用numpy

如果你的数据不是pandas数据框,那么可以使用numpy数组来将列转换为列表。首先,你需要将数据转换为numpy数组。例如,假设我们有以下姓名和年龄列表:

import numpy as np

names = ["Alice", "Bob", "Charlie"]
ages = [25, 30, 35]

# 将数据转换为numpy数组
data = np.array([names, ages])

要将年龄列转换为列表,可以使用以下代码:

age_list = data[1].tolist()

这将生成一个名为age_list的列表,其中包含了numpy数组中索引为1的行的所有数据。

使用列表推导式

最后,你也可以使用列表推导式来将数据框中的列转换为列表。以下代码将返回数据框中"age"列的所有数据:

age_list = [row["age"] for index, row in df.iterrows()]

这将生成一个名为age_list的列表,其中包含了数据框中"age"列的所有数据。

这是一种非常灵活的方法,因为你可以根据需要定制列表推导式来从数据框中选择任意列的数据。

以上是将数据框中的列转换为列表的几种方法,你可以根据自己的需求选择其中的一种。