优化代码 data = pd.concat([dates_year, dates_ratings], axis=1) data.columns = ['year', 'rating'] data = data.groupby('year').mean().reset_index() data['rating'] = data['rating'].round(1)
时间: 2023-08-08 13:13:21 浏览: 99
优化算法的代码
可以将这几行代码优化成一行:
```
data = pd.concat([dates_year, dates_ratings], axis=1).groupby('year').mean().round(1).reset_index().rename(columns={'rating': 'mean_rating'})
```
这样就不需要多次对 `data` 进行操作,而是一次性地使用 `groupby` 和 `mean` 函数完成了对年份的分组并计算了平均评分,然后使用 `round` 函数将结果保留一位小数,最后使用 `reset_index` 函数重置索引并将列名 `rating` 改为 `mean_rating`。
阅读全文