Python | 熊猫索引tolist()

简介

在Python中，pandas库提供了一个强大的数据操作工具，其中的DataFrame和Series对象具有许多有用的方法。tolist()是其中一个方法，它用于将DataFrame或Series对象转换为Python列表。

语法

DataFrame.to_list(self, n)   # n为可选参数,表示将列表展平为一维还是保留嵌套关系，0表示展平，1表示保留嵌套关系

返回值

tolist()方法返回一个由DataFrame或Series对象所有值组成的列表。

示例

假设我们有一个DataFrame对象：

import pandas as pd

data = {'Name': ['Tom', 'John', 'Smith', 'Jessica'],
        'Age': [25, 28, 32, 27],
        'City': ['New York', 'London', 'Paris', 'Tokyo']}
df = pd.DataFrame(data)

我们可以使用tolist()方法将DataFrame对象转换为一个列表：

df_list = df['City'].tolist()

得到的列表将是：

['New York', 'London', 'Paris', 'Tokyo']

此时的df_list即为转换后的列表。如果我们传递参数n=1，将列表展平为一维，则得到：

[['New York'], ['London'], ['Paris'], ['Tokyo']]

注意事项

• 如果DataFrame对象的某一列数据类型不一致，则转换为列表时，所有元素都以最宽泛的类型来表示。
• 如果Series对象包含缺失值（NaN），则它们将显示为列表中的浮点值nan。

小结

tolist()方法是pandas库中非常有用的功能之一，它允许我们将DataFrame或Series对象转换为Python列表。这对于进行某些数据操作、传递数据给其他函数或库来说非常方便。