假设我们有一个名为“data”的数据框,其中包含名为“group”的列和名为“value”的列。我们希望按组分组并将数据分类到一个名为“group_data”的字典中。每个键将是一个组名称,每个键的值将是该组的值列表。
可以使用pandas的groupby方法来按组分组,然后使用字典推导式来分类数据。
代码如下:
import pandas as pd
# 创建示例数据框
data = pd.DataFrame({'group': ['A', 'A', 'B', 'B', 'B', 'C'], 'value': [1, 2, 3, 4, 5, 6]})
# 按组分组并分类数据
group_data = {group: list(data.loc[data['group'] == group, 'value']) for group in data['group'].unique()}
# 打印结果
print(group_data)
输出:
{'A': [1, 2], 'B': [3, 4, 5], 'C': [6]}
在此示例中,我们首先创建了一个名为“data”的数据框,其中包含名为“group”的列和名为“value”的列。然后,我们使用data['group'].unique()获取组的唯一值列表,并使用此列表的每个元素来进行分组。groupby方法将数据框按照“group”列进行分组。我们然后使用字典推导式,将每个组的值列表添加到字典中。
我们得到的group_data字典是一个以组名为键,以值列表为值的字典。我们可以使用这个字典来进一步操作和分析我们的数据。
上一篇:按组合创造长格式
下一篇:按组和间隔坐标折叠数据框