下面是一个使用Pandas库进行数据分组并保留值的示例代码:
import pandas as pd
# 创建示例数据
data = {'Group': ['A', 'A', 'B', 'B', 'C', 'C'],
'Value': [1, 2, 3, 4, 5, 6]}
df = pd.DataFrame(data)
# 按照Group列进行分组,并保留每个分组的所有值
grouped_df = df.groupby('Group')['Value'].apply(list).reset_index()
print(grouped_df)
输出结果如下:
Group Value
0 A [1, 2]
1 B [3, 4]
2 C [5, 6]
在上面的代码中,首先创建了一个包含Group和Value列的DataFrame。然后,使用groupby()
函数按照Group列进行分组,并使用apply(list)
函数将每个分组的Value列转换为列表。最后,使用reset_index()
函数重置索引,以得到最终的结果。
这样,每个分组的所有值都会以列表的形式保留在新的DataFrame中,可以根据需要进行后续的分析或处理。
上一篇:按照Pandas列分别设置样式