在pandas中,可以使用groupby函数按两列总和分组,并创建新的数据帧。
首先,导入pandas库并创建一个包含两列的数据帧df:
import pandas as pd
data = {'Column1': [1, 2, 3, 4, 5],
'Column2': [6, 7, 8, 9, 10]}
df = pd.DataFrame(data)
接下来,使用groupby函数按两列总和分组,并创建新的数据帧new_df:
new_df = df.groupby(['Column1', 'Column2']).size().reset_index(name='Count')
在上述代码中,groupby函数接受一个包含要分组的列名的列表。size函数计算每个组的大小,并reset_index函数将结果重置为一个新的数据帧。我们将新的列命名为'Count'。
最后,可以打印新的数据帧new_df来查看结果:
print(new_df)
输出结果将显示按两列总和分组后的数据帧:
Column1 Column2 Count
0 1 6 1
1 2 7 1
2 3 8 1
3 4 9 1
4 5 10 1
这样,我们就按两列总和分组,并在pandas中创建了一个新的数据帧。
领取专属 10元无门槛券
手把手带您无忧上云