如何对 Pandas DataFrame 行进行分组并将列值转换为列表？

对 DataFrame 行进行分组以在 GroupBy 中创建列表的方法

在使用 pandas 进行数据操作的领域中，通常需要操作 DataFrame 行成特定格式。一个常见的要求是根据特定列对行进行分组，并以列表的形式从另一列检索值。

考虑一个包含“a”和“b”列的 DataFrame，如下所示：

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

目标是将此 DataFrame 转换为一个新的 DataFrame，其中行按列“a”分组，列“b”中的值转换为每个组的列表。所需的输出如下所示：

A [1, 2]
B [5, 5, 4]
C [6]

为了实现这一点，我们可以利用 pandas 的 'groupby' 和 'apply' 函数，如下所示：

# Import the pandas library
import pandas as pd

# Create a DataFrame from the provided data
df = pd.DataFrame( {'a':['A','A','B','B','B','C'], 'b':[1,2,5,5,4,6]})

# Group the DataFrame by column 'a'
grouped = df.groupby('a')

# Apply the list function to each group to get the 'b' values as lists
group_b_lists = grouped['b'].apply(list)

# Reset the index of the resulting Series to obtain a DataFrame
df_result = group_b_lists.reset_index(name='b_lists')

# Print the transformed DataFrame
print(df_result)

此代码有效按列“a”对原始 DataFrame 进行分组，将列表函数应用于每个组，并将结果列表分配给名为“b_lists”的新列。然后打印生成的 DataFrame 以显示所需的输出。

以上是如何对 Pandas DataFrame 行进行分组并将列值转换为列表？的详细内容。更多信息请关注PHP中文网其他相关文章！