diff --git a/q4-1.py b/q4-1.py new file mode 100644 index 0000000..dbd889b --- /dev/null +++ b/q4-1.py @@ -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() \ No newline at end of file