从pandas数据帧的列或行获取列表

摘要

本教程将介绍如何从pandas数据帧的列或行中获取列表。在使用pandas进行数据处理和分析时，经常需要将数据帧的特定列或行提取为列表，以便进行进一步的操作。本教程将给出几种方法来实现从数据帧提取列表的操作。

内容

在pandas中，可以使用以下几种方法从数据帧的列或行中获取列表：

方法一：使用tolist()方法

使用tolist()方法可以将pandas的Series对象转换为Python列表。例如：

 1import pandas as pd
 2
 3data_dict = {'one': pd.Series([1, 2, 3], index=['a', 'b', 'c']),
 4             'two': pd.Series([1, 2, 3, 4], index=['a', 'b', 'c', 'd'])}
 5
 6df = pd.DataFrame(data_dict)
 7
 8column_one_list = df['one'].tolist()
 9
10print(column_one_list)

方法二：使用values属性

1column_one_list = df['one'].values.tolist()

在上述示例中，我们使用values属性获取列的值，并使用tolist()方法将其转换为列表。

方法三：使用values属性和reshape()方法

1import numpy as np
2
3column_one_list = df['one'].values.reshape(1, -1).ravel().tolist()

在上述示例中，我们使用values属性获取列的值，并使用reshape()方法将其重塑为numpy数组，然后使用ravel()方法展平数组，并最后使用tolist()方法将其转换为列表。

无论选择哪种方法，都可以从数据帧的列或行中获取列表，并进行后续的操作。

总结

在pandas中，我们可以使用tolist()方法或values属性配合tolist()方法来从数据帧的列或行中获取列表。根据实际需求选择合适的方法进行操作。