tolist函数的使用方法

tolist 函数

tolist 函数是 Pandas 库中的一个函数，作用是将 DataFrame 或者 Series 类型的数据转换为 Python 的嵌套列表或者列表。它支持一维或者多维数组的转换，并且可以通过参数控制返回值的类型。具体的函数定义如下：

    DataFrame.tolist()
    Series.tolist()

tolist 函数不接受参数，只需要直接调用即可。下面我们对这个函数的两个具体用法进行介绍。

1. 将 DataFrame 转换为列表

我们先来创建一个 DataFrame，然后将其转换为列表：

    import pandas as pd

    df=pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
    res=df.tolist()
    print(res)
    

运行上面的代码，输出结果为:

    [[1, 4], [2, 5], [3, 6]]
    

可以看到，tolist 函数将 DataFrame 转换成了一个嵌套列表。其中，每个子列表对应了 DataFrame 中的一行数据。

2. 将 Series 转换为列表

我们再来看一下将 Series 转换为列表的例子。同样的，我们先创建一个 Series，然后通过 tolist 函数将其转换成列表：

    import pandas as pd

    s=pd.Series([1, 2, 3])
    res=s.tolist()
    print(res)
    

运行上面代码，输出结果为：

    [1, 2, 3]
    

可以看到，此时 tolist 将 Series 直接转换成了列表。

总结

tolist() 可以将 DataFrame 或 Series 类型的数据转换为 Python 的嵌套列表或者列表。它支持一维或者多维数组的转换，并且可以通过参数控制返回值的类型。