Pandas 的'groupby”函数如何有效地从分组行创建值列表？

通过 Groupby 在 Pandas 中高效地将行分组为列表

对于数据操作任务，根据特定条件将行分组为列表是常见的要求。在 Pandas 中，groupby 函数为此目的提供了一个强大的工具。

假设您有一个包含两列“a”和“b”的 DataFrame：

a b
A 1
A 2
B 5
B 5
B 4
C 6

目标是基于“a”列对行进行分组，并为每个组创建“b”列的列表。

要实现此目的，您可以利用groupby 函数：

df.groupby('a')['b'].apply(list)

groupby 函数按“a”列对 DataFrame 进行分组。然后，apply 函数迭代每个组，并使用 list 将“b”列转换为列表。

结果输出：

a
A       [1, 2]
B    [5, 5, 4]
C          [6]
Name: b, dtype: object

此技术使您能够根据特定列并获取每组中其他列的值列表。

以上是Pandas 的'groupby”函数如何有效地从分组行创建值列表？的详细内容。更多信息请关注PHP中文网其他相关文章！