完成q4_1:电影类型柱状图,包含代码、数据和图片
This commit is contained in:
45
q4/q4_1/q4_1.py
Normal file
45
q4/q4_1/q4_1.py
Normal file
@@ -0,0 +1,45 @@
|
||||
import json
|
||||
import matplotlib.pyplot as plt
|
||||
from collections import Counter
|
||||
|
||||
# 设置中文字体(解决中文乱码问题,必加!)
|
||||
plt.rcParams['font.sans-serif'] = ['SimHei'] # 黑体
|
||||
plt.rcParams['axes.unicode_minus'] = False # 解决负号显示问题
|
||||
|
||||
# 1. 读取movies.json数据
|
||||
try:
|
||||
with open('movies.json', 'r', encoding='utf-8') as f:
|
||||
movies_data = json.load(f)
|
||||
except FileNotFoundError:
|
||||
print("错误:未找到movies.json文件,请确认文件在当前目录!")
|
||||
exit()
|
||||
|
||||
# 2. 统计各类型电影数量(处理多类型分隔,比如"剧情/喜剧"拆分为两个类型)
|
||||
genre_list = []
|
||||
for movie in movies_data:
|
||||
# 假设movies.json里的类型字段是"genre",如果是其他名称(如"type")请修改
|
||||
genres = movie.get('genre', '').split('/') # 按/拆分多类型
|
||||
genre_list.extend([g.strip() for g in genres if g.strip()]) # 去空格并过滤空值
|
||||
|
||||
genre_counter = Counter(genre_list) # 统计各类型数量
|
||||
|
||||
# 3. 提取X轴(类型)和Y轴(数量)数据
|
||||
x_labels = list(genre_counter.keys())
|
||||
y_values = list(genre_counter.values())
|
||||
|
||||
# 4. 绘制柱状图
|
||||
plt.figure(figsize=(10, 6)) # 设置画布大小
|
||||
plt.bar(x_labels, y_values, color='skyblue') # 用plt.bar绘制柱状图
|
||||
|
||||
# 5. 设置标题和坐标轴(按要求配置)
|
||||
plt.title("类型电影数量分布", fontsize=14) # 标题
|
||||
plt.xlabel("电影类型", fontsize=12) # X轴标签(可选,更清晰)
|
||||
plt.ylabel("电影数量", fontsize=12) # Y轴标签(可选,更清晰)
|
||||
plt.xticks(rotation=45, ha='right') # X轴文字旋转,避免重叠(可选优化)
|
||||
plt.tight_layout() # 自动调整布局,防止文字被截断
|
||||
|
||||
# 6. 保存图片(按要求:q4_1_bar.png,dpi=150)
|
||||
plt.savefig('q4_1_bar.png', dpi=150)
|
||||
plt.close() # 关闭画布,释放资源
|
||||
|
||||
print("柱状图绘制完成!已保存为q4_1_bar.png")
|
||||
Reference in New Issue
Block a user