首页 > 其他分享 >数据挖掘——matplotlib

数据挖掘——matplotlib

时间:2024-07-11 21:00:52浏览次数:15  
标签:set movie axes 百万美元 matplotlib plt 数据挖掘

matplotlib概述

Mat指的是Matlab,plot指的是画图,lib即library,顾名思义,matplotlib是python专门用于开发2D图表的第三方库,使用之前需要下载该库,使用pip命令即可下载。

pip install matplotlib

1、matplotlib画图流程

matplotlib可分为三层结构,分别为:容器层(canvas),辅助显示层和显示层,而容器显示层又可以分为画板层,画布层和绘图区。辅助显示层主要是添加网格,图例等,增加可读性。而显示层则负责是将图像显现出来。

# 简单画图-折线图
import matplotlib.pyplot as plt
plt.figure()  # 容器层
plt.plot([1, 0, 9], [4, 5, 6])   # 三个点(1,4)(0,5)(9,6)  并连线
plt.show()  # 显示图像

结果如下,依次将三点连线
在这里插入图片描述

2、常用操作

(a)是否添加网格

plt.grid(visible,linstyle,alpha)
# visible传入布尔值,为True表示可见
# linestyle 表示线的形状
# alpha 表示透明系数

(b) 修改刻度

plt.xticks(ticks, [labels], **kwargs)
# ticks:数组类型,用于设置X轴刻度间隔
# [labels]:数组类型,用于设置每个间隔的显示标签
# **kwargs:用于设置标签字体倾斜度和颜色等外观属性。

(c )设置x,y轴说明以及标题

plt.xlabel('name')   # 设置x轴名称
plt.ylabel('name')   # 设置y轴名称
plt.title('title')   # 设置标题

(d)设置图例

plt.legend(loc='best')  # loc指的为图例位置,还有其他取值,不过一般设为best即可

(e) 设置多个绘图区
如果想要在一幅图上显示多个图像,可以使用如下方法:

figure,axes = plt.subplots(nrows,ncols)
# 划分绘图区,参数表示几行几列,默认一行一列,即一个绘图区
# 其返回值为一个figure对象和axes,使用axes[i]可以操作对应画图
# 设置每个子图区和上面大部分操作相同,但是对于修改x,y轴标题和刻度的时候需要修改一下方法名
# 修改对应的刻度使用axes[i].set_xticks(),修改x,y轴说明和标题用set_xlabel(),set_ylabel(),set_title()

(f)设置图片大小

plt.figure(figsize,dpi) 
# figsize传入一个元组,指定figure的宽和高,单位为英寸
# dpi指定绘图对象的分辨率,即每英寸多少个像素,缺省值为80

(g)图片的保存

plt.savefig(loc)
# loc指存储路径,默认为当前目录
# 注意保存必须在plt.show()之前进行,否则保存出来是一个空白图片

3、常用的图形(折线图,柱状图,饼状图,直方图,散点图)

(a) 折线图(plot)

 plt.plot(x,y,color,linestyle,label)
 """
 x:横轴数据
 y:纵轴数据
 color:颜色,可以使用RGB或者英语单词
 linestyle:线的类型
 label:标签,如果后面设置图例,需要在此设置label
 """

需求:利用折线图绘制上海和北京两地上午11点到12点的气温变化,要求创建两个绘图区并显示图例,加上网格以及x轴、y轴说明,x轴时间刻度为每5分钟显示一次,y轴显示气温为0到40之间,一小格代表5摄氏度。上海温度在25-28摄氏度之间,北京温度在20-27摄氏度之间。

import matplotlib.pyplot as plt
import random
plt.rcParams['font.family'] = 'STFangsong'  # 正常显示中文:仿宋体
# 准备数据
x = range(60)
y_shanghai = [random.uniform(25, 28) for i in x]
y_beijing = [random.uniform(20, 27) for i in x]
figure, axes = plt.subplots(nrows=1, ncols=2, figsize=(20, 8), dpi=80)   # 创建两个绘图区,一行两列
axes[0].plot(x, y_shanghai, color='red', label='上海')
axes[1].plot(x, y_beijing, color='green', linestyle='--', label='北京')
# 设置刻度
time = ['11点{}分'.format(i) for i in range(0, 60, 5)]
axes[0].set_xticks(x[::5], time)
axes[1].set_xticks(x[::5], time)
# 设置y轴刻度
axes[0].set_yticks(range(0, 40, 5))
axes[1].set_yticks(range(0, 40, 5))
# 设置x,y轴说明
axes[0].set_xlabel("时间")
axes[1].set_xlabel("时间")
axes[0].set_ylabel("温度")
axes[1].set_ylabel("温度")
# 设置图例,和前面plot中的label对应
axes[0].legend()
axes[1].legend()
# 加入网格,透明度为0.5
axes[0].grid(True, linestyle='--', alpha=0.5)
axes[1].grid(True, linestyle='--', alpha=0.5)
# 加标题
axes[0].set_title("上海中午11点到12点气温折线图")
axes[1].set_title("北京中午11点到12点气温折线图")
plt.show()

示意图如下所示:
折线图实例
(b)散点图(scatter)

plt.scatter(x,y)
"""
x:横轴数据
y:纵轴数据
散点图只在坐标系上描出对应的点,可以看出变量之间的关系
"""

需求:用散点图画出房价和面积之间的关系,数据如下:房价(万元):[195.53, 203.88, 210.75, 372.74, 202.41, 247.61, 24.9, 239.34, 140.32, 104.15, 176.84, 288.23, 128.79, 49.64, 191.74, 33.1, 30.74, 400.02,205.35, 330.64, 283.45],对应的面积(平方米):[225.98, 247.97, 253.14, 457.85, 241.58, 301.01, 20.67, 288.64, 163.56,120.06, 207.83, 342.75, 147.9, 53.06, 224.72, 29.51, 21.61, 483.21,245.25, 399.25, 343.35]

import matplotlib.pyplot as plt
plt.rcParams['font.family'] = 'STFangsong'
# 创建画布
plt.figure(figsize=(20, 8), dpi=80)
# 绘制图形
x = [225.98, 247.97, 253.14, 457.85, 241.58, 301.01, 20.67, 288.64, 163.56,
     120.06, 207.83, 342.75, 147.9, 53.06, 224.72, 29.51, 21.61, 483.21,
     245.25, 399.25, 343.35]
y = [195.53, 203.88, 210.75, 372.74, 202.41, 247.61, 24.9, 239.34, 140.32,
     104.15, 176.84, 288.23, 128.79, 49.64, 191.74, 33.1, 30.74, 400.02,
     205.35, 330.64, 283.45]
plt.scatter(x, y)
plt.xlabel("房屋面积(平方米)")
plt.ylabel("房屋价格(万元)")
plt.xticks(range(0, 500, 50))
plt.show()

示意图如下所示:
在这里插入图片描述
经图像分析,房屋面积和房屋价格呈正相关。
(c )柱状图(Bar)

plt.bar(x,y,color)
# x 表示x轴数据
# y 表示y轴数据
# color 表示颜色,可以传入一个列表

需求:画出电影与对应票房的柱状图
movie = [“头脑特工队2”,
“沙丘:第二部分”,
“哥斯拉大战金刚:新帝国”,
“功夫熊猫4”,
“猩球崛起:王国”,
“小丑2:双重疯狂”,
“捉鬼敢死队:冰冻帝国”,
“鲍勃·马利:一爱”,
“养蜂人”,
“木法沙:狮子王”
]
data = [
850.5, # 百万美元(预测)
720, # 百万美元(预测)
650, # 百万美元(预测)
600, # 百万美元(预测)
450, # 百万美元(估计)
800, # 百万美元(预测)
188, # 百万美元
177, # 百万美元
153, # 百万美元
690 # 百万美元(预测)
]

import matplotlib.pyplot as plt
plt.rcParams['font.family'] = 'STFangsong'
movie = [
    "头脑特工队2",
    "沙丘:第二部分",
    "哥斯拉大战金刚:新帝国",
    "功夫熊猫4",
    "猩球崛起:王国",
    "小丑2:双重疯狂",
    "捉鬼敢死队:冰冻帝国",
    "鲍勃·马利:一爱",
    "养蜂人",
    "木法沙:狮子王"
]
data = [
    850.5,  # 百万美元(预测)
    720,    # 百万美元(预测)
    650,    # 百万美元(预测)
    600,    # 百万美元(预测)
    450,    # 百万美元(估计)
    800,    # 百万美元(预测)
    188,    # 百万美元
    177,    # 百万美元
    153,    # 百万美元
    690     # 百万美元(预测)
]
plt.figure(figsize=(20, 8), dpi=80)
color = ['b', 'r', 'g', 'y', 'c', 'm', 'y', 'k', 'c', 'g']
plt.bar(movie, data, color=color)
plt.grid(True, linestyle="--", alpha=0.5)
plt.title("电影票房")
plt.xlabel("电影名称")
plt.ylabel("票房(百万美元)")
plt.show()

示意图如下所示:
电影票房柱状图
(d)直方图(hist)
直方图与柱状图的区别为:直方图各个柱状之间是连续的,而柱状图是离散的。

plt.hist(x,bins,**kwargs(density)) 
# x为数据,bins为组数,density为频率,为True则显示为频率

需求:描绘电影时长的频率分布直方图,时长数据见具体程序。

import matplotlib.pyplot as plt
plt.rcParams['font.family']='STFangsong'
movie_time = [131, 98, 125, 131, 124, 139, 131, 117, 128, 108, 135, 138, 131, 102, 107, 114, 119, 128, 121, 142, 127,
              130, 124, 101, 110, 116, 117, 110, 128, 128, 115, 99, 136, 126, 134, 95, 138, 117, 111, 78, 132, 124, 113,
              150, 110, 117, 86, 95, 144, 105, 126, 130, 126, 130, 126, 116, 123, 106, 112, 138, 123, 86, 101, 99, 136,
              123, 117, 119, 105, 137, 123, 128, 125, 104, 109, 134, 125, 127, 105, 120, 107, 129, 116, 108, 132, 103,
              136, 118, 102, 120, 114, 105, 115, 132, 145, 119, 121, 112, 139, 125, 138, 109, 132, 134, 156, 106, 117,
              127, 144, 139, 139, 119, 140, 83, 110, 102, 123, 107, 143, 115, 136, 118, 139, 123, 112, 118, 125, 109,
              119, 133, 112, 114, 122, 109, 106, 123, 116, 131, 127, 115, 118, 112, 135, 115, 146, 137, 116, 103, 144,
              83, 123, 111, 110, 111, 100, 154, 136, 100, 118, 119, 133, 134, 106, 129, 126, 110, 111, 109, 141, 120,
              117, 106, 149, 122, 122, 110, 118, 127, 121, 114, 125, 126, 114, 140, 103, 130, 141, 117, 106, 114, 121,
              114, 133, 137, 92, 121, 112, 146, 97, 137, 105, 98, 117, 112, 81, 97, 139, 113, 134, 106, 144, 110, 137,
              137, 111, 104, 117, 100, 111, 101, 110, 105, 129, 137, 112, 120, 113, 133, 112, 83,
              94, 146, 133, 101, 131, 116, 111, 84, 137, 115, 122, 106, 144, 109, 123, 116, 111, 111, 133, 150]
plt.figure(figsize=(20, 8), dpi=80)
plt.hist(movie_time, bins=int((max(movie_time)-min(movie_time))//2), density=True)  # 频率分布直方图,所有density为True
plt.xticks([i for i in range(min(movie_time), max(movie_time)+2, 2)])   # 加2因为range是左闭右开,防止最大值取不到,所以加2
plt.grid(linestyle='--', alpha=0.5)
plt.title("电影时长分布直方图")
plt.xlabel("时长")
plt.ylabel("频率")
plt.show()

示意图如下所示:
电影时长频率分布直方图
(e)饼状图

plt.pie(x,labels,autopct,colors)
# x:数量,根据数量自动计算百分比
# labels:每部分名称,需要和x一一对应
# autopct:占比指定%1.2f%%
# colors:每部分颜色
# 一般超过9个类的时候就不用饼图了,而使用柱形图

需求:描绘电影和对应票房的饼状图。
movie_name = [‘雷神3:诸神黄昏’, ‘正义联盟’, ‘东方快车谋杀案’, ‘寻梦环游记’, ‘全球风暴’, ‘降魔传’, ‘追捕’, ‘七十七天’, ‘密战’,‘狂兽’, ‘其它’]
place_count = [60605, 54546, 45819, 28243, 13270, 9945, 7679, 6799, 6101, 4621, 20105]

import matplotlib.pyplot as plt
plt.rcParams['font.family'] = 'STFangsong'
movie_name = ['雷神3:诸神黄昏', '正义联盟', '东方快车谋杀案', '寻梦环游记', '全球风暴', '降魔传', '追捕', '七十七天', '密战',
              '狂兽', '其它']
place_count = [60605, 54546, 45819, 28243, 13270, 9945, 7679, 6799, 6101, 4621, 20105]
plt.figure(figsize=(20, 8), dpi=80)
plt.pie(place_count, labels=movie_name, autopct="%1.2f%%", colors=['b', 'r', 'g', 'y', 'c', 'm', 'y', 'k', 'c', 'g', 'y'])
plt.legend()
# 将长径和短径设为相等,构成一个圆,否则为椭圆
plt.axis('equal')
plt.show()

示意图如下所示:
在这里插入图片描述

4、总结

对于各类图形的绘画,发现只是在方法名和参数上有所不同,只需要掌握折线图对应plot,散点图对应scatter,柱状图对应bar,饼图对应pie,直方图对应hist,再传入具体的参数即可。
折线图:plt.plot(x,y) # x为横轴数据,y为纵轴数据
散点图:plt.scatter(x,y) # x为横轴数据,y为纵轴数据
柱状图:plt.bar(x,y) # x为横轴数据,y为纵轴数据
饼图:plt.pie(x,labels) # x轴数据,labels为对应的标签
直方图:plt.hist(x,bins,density) # x为数据,bins为组数,density为是否按照频率显示,默认为False

标签:set,movie,axes,百万美元,matplotlib,plt,数据挖掘
From: https://blog.csdn.net/weixin_50164178/article/details/140255261

相关文章

  • 数据挖掘习题9
    1.题干    AQIandLatLongofCountries.csv数据集的目标是为不同地区的空气质量提供有价值的见解,使研究人员和政策制定者能够就如何解决空气污染问题做出明智的决定。该数据集由两个独立的数据集合并而成,一个包含城市及其相应的经纬度坐标信息,另一个包含世界各国的......
  • [Python基础] matplotlib绘图的深入浅出学习
    matplotlib 是Python中最常用的绘图库之一,它提供了丰富的绘图功能,非常适合数据可视化。下面我将从整体逻辑开始,逐步深入到具体的例子matplotlib绘图整体逻辑:1、创建图像对象: plt.figure()2、绘制数据:plt.plot()等函数绘制数据 3、设置坐标轴、标签、图表标题等;现在,......
  • Python 数据挖掘
    数据挖掘基础数据挖掘:这种从数据中“淘金”,从大量数据(包括文本)中挖掘出隐含的、未知的、对决策有潜在价值的关系、模式和趋势,并用这些知识和规则建立用于决策支持的模型,提供预测性决策支持的方法、工具和过程,就是数据挖掘;它是利用各种分析工具在大量数据中寻......
  • matplotlib绘制拟合大象曲线
    通过十个参数绘制大象曲线importmatplotlib.pyplotaspltimportnumpyasnpimportmatplotlib.animationasanimationnum=101k=np.arange(1,6)t=np.linspace(0,2*np.pi,num)Ax=[0,0,12,0,-14]Bx=[50,18,0,0,0]Ay=[-60,0,0,0,0]By=[-30,8,-10,0,0]eye=20x=(np......
  • python:使用matplotlib库绘制图像(一)
    作者是跟着http://t.csdnimg.cn/4fVW0学习的,matplotlib系列文章是http://t.csdnimg.cn/4fVW0的自己学习过程中整理的详细说明版本,对小白更友好哦!一、Matplotlib图像基础1.1 基本绘图实例:sin、cos函数图代码详解:1.frompylabimport*:导入pylab库中所有函数和变量。pyla......
  • 数据挖掘小参考资料推荐
    原文链接博客github目录[判别式模型和生成式模型的区别|知乎](https://zhuanlan.zhihu.com/p/74586507)[Gini系数生成决策树|博客](https://victorzhou.com/blog/gini-impurity/)[一个学数据挖掘的开发者的博客|博客](https://victorzhou.com/page/5/)[信息增......
  • matplotlib画3D图形的contour和plot_surface函数参数
    importmatplotlib.pyplotaspltimportnumpyasnpfrommpl_toolkits.mplot3dimportAxes3Dfig=plt.figure()ax=fig.add_axes(Axes3D(fig))X=np.arange(-4,4,0.25)Y=np.arange(-4,4,0.25)X,Y=np.meshgrid(X,Y)R=np.sqrt(X**2+Y**2)Z=np.sin......
  • 【python数据挖掘案列】利用线性回归(LR)实现天气变化的时间序列预测
    利用线性回归实现天气变化的时间序列预测一、引言天气预测一直是气象学和机器学习领域的重要研究内容。时间序列预测是其中的一种常见任务,旨在通过分析历史数据来预测未来的天气情况。在本文中,我们将使用线性回归模型来实现天气变化的时间序列预测,并介绍整个预测流程。二......
  • python 进阶教程--matplotlib
    matplotlib3.1安装配置3.2Matplotlib快速入门3.3图形绘制线图散点图条形图饼图3.4风格样式使用内置样式创建自定义样式临时使用样式3.1安装配置Matplotlib是一个用于创建高质量图表的Python绘图库。在开始使用Matplotlib之前,需要先安装它。以下是安装......
  • Matplotlib基本绘图
    Matplotlib基本绘图1.折线图(1)基本绘图plt.plot(x,y)importmatplotlib.pyplotaspltx=[1,2,3,4]y=[12,11,14,13]plt.plot(x,y)plt.show()(2)定义样式A.线条样式a.线条颜色plt.plot(x,y,color=颜色)importmatplotlib.pyplotaspltx=[1,2,3......