python - 如何自动构建多个列表

Question

我有数百个数据帧，假设名称是df1，...，df250，我需要通过这些数据帧的一列来构建列表。平时我都是手动做的，但是现在数据太多了，而且容易出错

这就是我所做的

list1 = df1['customer_id'].tolist()
list2 = df2['customer_id'].tolist()
..
list250 = df250['customer_id'].tolist()

这太手动了，我们可以用更简单的方式来做吗？

Answer 1

更简单的方法是后退一步并确保将数据帧放入list或dict等集合中。然后，您可以以可扩展的方式轻松执行操作。

例如:

dfs = {1: df1, 2: df2, 3: df3, ... , 250: df250}

lists = {k: v['customer_id'].tolist() for k, v in dfs.items()}

然后您可以以 lists[1]、lists[2] 等形式访问结果。

还有其他好处。例如，您不再污染命名空间，您节省了显式定义变量名的精力，您可以轻松存储和传输相关的对象集合。