# -*- coding: utf-8 -*-
"""
Created on Mon Nov 14 22:05:20 2022
@author: GD005
"""
import pandas as pd
import matplotlib as mpl
import matplotlib.pyplot as plt
import numpy as np
# 读取数据
df = pd.read_excel('已清洗数据.xlsx')
# 提取电影时长 平均评分
run_time, rating_score = df['runtime'], df['vote_average']
# 设置中文显示
mpl.rcParams['font.family'] = 'SimHei'
# 设置图形显示风格
plt.style.use('ggplot')
# 设置大小 像素
plt.figure(figsize=(9, 6), dpi=100)
# 绘制散点图
plt.scatter(run_time, rating_score, c='purple')
plt.yticks(np.arange(0, 10.5, 1))
# 添加描述信息
plt.title('电影平均评分与电影时长的关系', fontsize=18, x=0.5, y=1.02)
plt.xlabel('电影时长(分钟)')
plt.ylabel('平均评分')
# 保存图片
plt.savefig('test_006.png')
# 显示图片
plt.show()
这段代码的功能是读取一个名为'已清洗数据.xlsx'的Excel文件,提取电影时长和平均评分数据,并使用Matplotlib库绘制散点图来展示电影平均评分与电影时长之间的关系,并将图表保存为test_006.png文件。
在代码中,首先使用pandas库读取了Excel文件中的数据,然后提取了电影时长和平均评分数据。接着使用Matplotlib库设置了中文显示和图形显示风格,然后创建了一个散点图,展示了电影平均评分与电影时长之间的关系。最后将图表保存为test_006.png文件,并显示出来。