index = pd.MultiIndex.from_product([Order_df['Arrive_step'].unique(), Order_df['Pickup_Zone'].unique(), Order_df['Dropoff_Zone'].unique()], names=['Arrive_step', 'Pickup_Zone', 'Dropoff_Zone']) grouped_Travel_time = Order_df.groupby(['Arrive_step', 'Pickup_Zone', 'Dropoff_Zone'])['Travel_time'].mean().reindex(index, fill_value=Order_df.groupby(['Pickup_Zone', 'Dropoff_Zone'])['Travel_time'].mean()) grouped_Travel_time grouped_Travel_time.to_csv('./Data/grouped_Travel_time_20150114.csv')
时间: 2023-09-04 09:12:09 浏览: 71
这段代码使用了 Python 中的 Pandas 库,通过对 Order_df 数据集进行分组和处理,得到了一个多层次索引的数据表 grouped_Travel_time,并将其保存为 CSV 文件。其中,通过 pd.MultiIndex.from_product() 方法生成了一个由 Arrive_step、Pickup_Zone 和 Dropoff_Zone 三个变量构成的多层次索引,然后使用 groupby() 方法对这些变量进行分组,并计算 Travel_time 的平均值,最后使用 reindex() 方法将得到的结果根据多层次索引进行重构,同时使用 fill_value 参数指定了当某些组合不存在时,使用另一个分组的平均值进行填充。最后将得到的数据表使用 to_csv() 方法保存为 CSV 文件。
相关问题
index = pd.MultiIndex.from_product([Order_df['Arrive_step'].unique(), Order_df['Pickup_Zone'].unique(), Order_df['Dropoff_Zone'].unique()], names=['Arrive_step', 'Pickup_Zone', 'Dropoff_Zone']) grouped_Travel_time = Order_df.groupby(['Arrive_step', 'Pickup_Zone', 'Dropoff_Zone'])['Travel_time'].mean().reindex(index, fill_value=0) grouped_Travel_time
这段代码是用于对一个名为`Order_df`的数据框按照`Arrive_step`、`Pickup_Zone`和`Dropoff_Zone`三个列进行分组,并计算每组中`Travel_time`列的平均值。然后,使用`pd.MultiIndex.from_product()`方法,创建一个三级行索引,分别对应`Arrive_step`、`Pickup_Zone`和`Dropoff_Zone`三个列的唯一值。接着,使用`reindex()`方法将分组后的平均值数据按照新创建的行索引进行重构,并填充缺失值为0。最终,得到一个以三个列为索引,平均旅行时间为值的数据框。
# 用于填充缺失值的均值计算函数 def fill_missing_mean(group): mean = group.mean() return group.fillna(mean) # 构建多级索引 index = pd.MultiIndex.from_product([Order_df['Arrive_step'].unique(), Order_df['Pickup_Zone'].unique(), Order_df['Dropoff_Zone'].unique()], names=['Arrive_step', 'Pickup_Zone', 'Dropoff_Zone']) # 按照多级索引分组并计算均值,然后用均值填充缺失值 grouped_Travel_time = Order_df.groupby(['Arrive_step', Order_df['Pickup_Zone'], 'Dropoff_Zone']).apply(fill_missing_mean).groupby(['Arrive_step', 'Pickup_Zone', 'Dropoff_Zone'])['Travel_time'].mean().reindex(index, fill_value=0)那你尝试根据以上报错修改一下这个代码
根据报错信息,您可以尝试修改代码如下:
```
# 用于填充缺失值的均值计算函数
def fill_missing_mean(group):
mean = group.mean()
return group.fillna(mean)
# 构建多级索引
index = pd.MultiIndex.from_product([Order_df['Arrive_step'].unique(), Order_df['Pickup_Zone'].unique(), Order_df['Dropoff_Zone'].unique()], names=['Arrive_step', 'Pickup_Zone', 'Dropoff_Zone'])
# 按照多级索引分组并计算均值,然后用均值填充缺失值
grouped_Travel_time = Order_df.groupby(['Arrive_step', 'Pickup_Zone', 'Dropoff_Zone']).apply(fill_missing_mean).groupby(['Arrive_step', 'Pickup_Zone', 'Dropoff_Zone'])['Travel_time'].mean().reindex(index, fill_value=0)
```
主要是修改了 `groupby()` 方法的参数,将 `Order_df['Pickup_Zone']` 改为 `'Pickup_Zone'`,将 `Order_df['Dropoff_Zone']` 改为 `'Dropoff_Zone'`。同时,在 `reindex()` 方法中使用 `fillna()` 方法填充缺失值,而不是使用 `fill_value` 参数。
阅读全文