Python 在数据框中基于groupby从透视表创建OD矩阵

我正在尝试根据数据创建一个起点-终点矩阵 框架是从数据框架“按月分组”列中提取创建的。 我得到错误“keyrerror:'count'” 代码如下：

按月列对原始数据帧进行分组。 对于按每月最高计数排序的每个组，选择前10000行 创建范围为60000 x 60000的OD矩阵，行=从城市，列=到城市，值=计数 输出：

KeyError:“计数”

我如何修复错误？我漏了什么申报单吗

group_df=clean_df.groupby('month',axis='index').apply(lambda x: x.sort_values(by=['count'],ascending=False)).reset_index(drop=True)

sub_group_df=group_df.groupby('month',group_keys=False).apply(lambda g: g.nlargest(10000,"count"))

od_pivot = pd.DataFrame(sub_group_df, index=list(range(0,60000)), columns = list(range(0,60000))).add(sub_group_df.pivot_table(index="from_municipality",columns="to_municipality",values="count",aggfunc=len), fill_value=0).astype('int')
od_pivot