如何使用pandas统计转换后的列数据?
统计转换列的数据
想要统计转换列的数据,可以使用 pandas 库中的 get_dummies() 函数将分类变量转换为虚拟列,然后使用 groupby() 和 sum() 函数进行分组和求和。
以下代码展示了此过程:
import pandas as pddf = pd.dataframe({ 'date': ['2024-01-01', '2024-01-01', '2024-01-01', '2024-01-02', '2024-01-02', '2024-01-02', '2024-01-02', '2024-01-02', '2024-01-03', '2024-01-03', '2024-01-03', '2024-01-03'], 'type': [1, 2, 1, 3, 2, 3, 1, 1, 1, 4, 2, 5]})df_dummies = pd.get_dummies(df, columns=['type'])df_group = df_dummies.groupby("date").sum()print(df_dummies)print("-" * 60)print(df_group)
输出结果:
date type_1 type_2 type_3 type_4 type_50 2024-01-01 1 0 0 0 01 2024-01-01 0 1 0 0 02 2024-01-01 1 0 0 0 03 2024-01-02 0 0 1 0 04 2024-01-02 0 1 0 0 05 2024-01-02 0 0 1 0 06 2024-01-02 1 0 0 0 07 2024-01-02 1 0 0 0 08 2024-01-03 1 0 0 0 09 2024-01-03 0 0 0 1 010 2024-01-03 0 1 0 0 011 2024-01-03 0 0 0 0 1------------------------------------------------------------ type_1 type_2 type_3 type_4 type_5 date 2024-01-01 2 1 0 0 0 2024-01-02 2 1 2 0 0 2024-01-03 1 1 0 1 1