添加 q4-1.py
This commit is contained in:
28
q4-1.py
Normal file
28
q4-1.py
Normal 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()
|
||||
Reference in New Issue
Block a user