博客

pandas DataFrame数据转为list的方法

  • Post category:Python

Pandas DataFrame数据转为List的方法

在Python中，Pandas是一个常用的数据处理库，它提供了DataFrame数据结构来处理和分析数据。有时候，我们需要将DataFrame数据转换为List类型，以便于进行其他操作。本攻略将介绍Pandas DataFrame数据转为List的方法，包括使用values属性和to_numpy()方法。

使用values属性

Pandas DataFrame的values属性可以将DataFrame数据转换为一个二维数组，然后我们可以将其转换为List类型。以下是一个示例代码，演示如何使用values属性将DataFrame数据转换为List：

import pandas as pd

# 创建DataFrame数据
data = {'name': ['Tom', 'Jack', 'Steve', 'Ricky'], 'age': [28, 34, 29, 42]}
df = pd.DataFrame(data)

# 将DataFrame数据转换为List
list_data = df.values.tolist()

print(list_data)

在上面的示例代码中，我们首先创建了一个包含两个列的DataFrame数据。然后，我们使用values属性将DataFrame数据转换为一个二维数组，并使用tolist()方法将其转换为List类型。最后，我们输出了转换后的List数据。

需要注意的是，使用values属性将DataFrame数据转换为List时，List中的元素类型为numpy.ndarray类型，而不是Python内置的List类型。

使用to_numpy()方法

Pandas DataFrame的to_numpy()方法可以将DataFrame数据转换为一个NumPy数组，然后我们可以将其转换为List类型。以下是一个示例代码，演示如何使用to_numpy()方法将DataFrame数据转换为List：

import pandas as pd

# 创建DataFrame数据
data = {'name': ['Tom',Jack', 'Steve', 'Ricky'], 'age': [28, 34, 29, 42]}
df = pd.DataFrame(data)

# 将DataFrame数据转换为List
list_data = df.to_numpy().tolist()

print(list_data)

在上面的示例代码中，我们首先创建了一个包含两个列的DataFrame数据。然后，我们使用to_numpy()方法将DataFrame数据转换为一个NumPy数组，并使用tolist()方法将其转换为List类型。最后，我们输出了转换后的List数据。

需要注意的是，使用to_numpy()方法将DataFrame数据转换为List时，List中的元素类型为Python内置的List类型，而不是numpy.ndarray类型。

示例

以下是另一个示例代码，演示如何将DataFrame数据中的某一列转换为List：

import pandas as pd

# 创建DataFrame数据
data = {'name': ['Tom', 'Jack', 'Steve', 'Ricky'], 'age': [28, 34, 29, 42]}
df = pd.DataFrame(data)

# 将DataFrame数据中的age列转换为List
list_age = df['age'].tolist()

print(list_age)

在上面的示例代码中，我们首先创建了一个包含两个列的DataFrame数据。然后，我们使用[‘age’]索引将DataFrame数据中的age列提取出来，并使用tolist()方法将其转换为List类型。最，我们输出了转换后的List数据。

总结

本攻略介绍了Pandas DataFrame数据转为List的方法，包括使用values属性和to_numpy()方法。需要根据具体的需求选择合适的方法，并注意转换后List中的元素类型。同时，我们还提供了两个示例代码，演示了如何将整个DataFrame数据和DataFrame数据中的某一列转换为List。

Tags: