添加 q4-1.py

This commit is contained in:
2026-07-05 18:35:55 +08:00
parent ddccfd7956
commit ebe20653bd

28
q4-1.py Normal file
View File

@@ -0,0 +1,28 @@
# 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()