豆瓣电影 Top250 类型矩形树图Python代码
时间: 2023-11-05 10:53:17 浏览: 171
python可视化图表案例-径向柱图与矩形树图
5星 · 资源好评率100%
以下是使用Python绘制豆瓣电影Top250类型矩形树图的示例代码:
```python
import matplotlib.pyplot as plt
import pandas as pd
import squarify
# 读取数据
df = pd.read_csv('https://raw.githubusercontent.com/chenjiandongx/douban/master/top250.csv')
# 计算每个类型的电影数量
type_counts = df['type'].str.split('/', expand=True).stack().value_counts()
# 绘制树图
squarify.plot(sizes=type_counts.values, label=type_counts.index, alpha=.8 )
plt.axis('off')
plt.show()
```
此代码需要使用`pandas`和`matplotlib`库,以及`squarify`库来绘制矩形树图。首先从CSV文件中读取数据,并计算每个类型的电影数量。然后使用`squarify`库的`plot`函数绘制矩形树图,其中`sizes`参数为每个类型的电影数量,`label`参数为类型名称,`alpha`参数为透明度。最后使用`matplotlib`库的`axis`函数关闭坐标轴,使用`show`函数显示图像。
阅读全文