pythondataframe筛选列表的值转为list【常用】

pythondataframe筛选列表的值转为list【常用】

在Python中，使用Pandas库可以进行数据框DataFrame操作。有时我们需要从DataFrame中筛选出符合某些条件的数据，并将这些数据转为一个列表，方便后续的处理。本文将演示如何使用Pandas来实现这个功能。

数据准备

我们先定义一个简单的DataFrame，包含商品名称、单价和库存量三列数据。代码如下：

import pandas as pd
data = {'商品名称': ['商品A', '商品B', '商品C', '商品D', '商品E'],
        '单价': [10, 20, 30, 40, 50],
        '库存量': [100, 200, 300, 400, 500]}
df = pd.DataFrame(data)
print(df)

输出结果如下：

筛选列表的值转为list

假设我们要筛选出所有单价大于20的商品名称，代码如下：

result = df.loc[df['单价'] > 20, '商品名称'].tolist()
print(result)

输出结果如下：

['商品B', '商品C', '商品D', '商品E']

上面的代码中，我们使用了loc方法来筛选符合条件的数据。loc方法的第一个参数是筛选条件，可以是单个条件，也可以是多个条件的组合。本例中我们使用了单个条件：单价大于20。第二个参数为需要返回的列，本例中是'商品名称'。最后我们将返回的结果使用tolist()方法转为了列表。

总结

通过上面的演示，能够看出这种使用loc方法配合tolist方法的方式非常方便，简洁实用，成为了DataFrame中筛选数据的常用方式之一。当然，这并不是唯一的方式，如果您有其他更好的方法，也欢迎分享。