# q4/q4_1/q4_1.py
import json
import matplotlib.pyplot as plt

# 读取电影数据
with open("../q2_1_crawler/movies.json", "r", encoding="utf-8") as f:
    data = json.load(f)
movies = data["movies"]

# 统计类型数量
genre_dict = {}
for m in movies:
    g = m["genre"]
    genre_dict[g] = genre_dict.get(g, 0) + 1

x = list(genre_dict.keys())
y = list(genre_dict.values())

# 绘制柱状图
plt.bar(x, y)
plt.title("类型电影数量分布")
plt.xlabel("电影类型")
plt.ylabel("电影数量")
plt.tight_layout()  # 防止文字重叠

# 保存图片 dpi=150
plt.savefig("q4_1_bar.png", dpi=150)
plt.show()

打开 VS Code，顶部菜单栏点击 终端 (Terminal) → 新建终端 (New Terminal)
在底部弹出的终端窗口输入安装命令，回车执行
pip install matplotlib