pandas - pandas df 在 function 内创建 list 以在 function 外使用

我想在 df 的一列中创建前三个 values 的 list,但是这个 df 是在 function 中创建的,并且会使用不同的输入变量多次调用。每次我调用这个 function 时,我都希望将新的前三个添加到旧的前三个的 list 上。然后我希望能够在这个 function 之外使用这个 list,就像在输入 list 中调用不同的 function 一样。

所以在 function 中,第一次调用时,创建的 df 如下所示:

col1     col2
 A        1
 B        2
 C        3
 D        4

list 应该如下所示:

['A', 'B', 'C']

然后随着输入变量改变的下一次迭代, table 看起来像这样

col1     col2
 E        5
 F        6
 G        7
 H        8
 I        9

那么 list 应该如下所示:

['A', 'B', 'C', 'D', 'E', 'F']

那么我应该能够在这个 function 之外使用这个 list (作为不同 function 的输入)。有人可以帮我吗?在此先感谢您的帮助

回答1

您可以收集为 list 您的列

my_list = df[col1].tolist()

然后得到三个第一个元素

selected_items = my_list[:2]

然后与您之前的 list 连接

previous_list = previous_list + selected_items

显然之前的 list 应该被初始化。

您可以在流程的每次新迭代中执行这些流程。