PHP前端开发

如何使用pandas统计转换后的列数据?

百变鹏仔 5天前 #Python
文章标签 如何使用

统计转换列的数据

想要统计转换列的数据,可以使用 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