Python 使用 pivot 函数将 GroupBy 结果转为行列 DataFrame
您可以使用pivot函数来实现将'city'作为列,'pt'作为行的DataFrame。下面是一个示例代码:
import pandas as pd
# 假设有一个名为df的DataFrame,包含'city'和'pt'两列
# 使用groupby函数进行分组,并进行计数操作
df_grouped = df.groupby(['city', 'pt']).size().reset_index(name='count')
# 使用pivot函数将'city'作为列,'pt'作为行,'count'作为值来生成新的DataFrame
df_pivot = df_grouped.pivot(index='pt', columns='city', values='count')
在上述代码中,首先使用groupby函数对'city'和'pt'进行分组,并使用size函数进行计数操作,然后使用reset_index函数将结果重新设置为DataFrame。接着,使用pivot函数将'city'作为列,'pt'作为行,'count'作为值来生成新的DataFrame。最终,得到的df_pivot即为以'pt'为行,'city'为列的DataFrame。
原文地址: http://www.cveoy.top/t/topic/ii1e 著作权归作者所有。请勿转载和采集!