介绍

matplotlib库是Python中的一个2D绘图库。

安装

$ pip install matplotlib

入门

绘制折线图

pyplot函数绘制折线图

这里画一个月份与温度的折线图：

import matplotlib.pyplot as plt

x_month = [3,4,5,6,7,8,9]
y_temperature = [18,24,27,30,35,37,31]
plt.plot(x_month,y_temperature)
plt.show()
# plt.savefig('xxx.png') # 保存图标函数

这里导入pyplot模块并将其命名为plt。
show()用于展示图片。savefig()则用于保存图片。

设定x,y轴

import matplotlib.pyplot as plt

x_month = [1,2,3,4,5,6,7,8]
y_temperature = [11,22,23,43,20,23,42,42]
plt.plot(x_month,y_temperature,linewidth=5) # linewidth定义折线宽度
plt.title("This is Title") # 标题
plt.xlabel("month",fontsize=14) # x轴
plt.ylabel("temperature",fontsize=14) # y轴
plt.tick_params(axis='both',labelsize=14) # 影响xy刻度的函数
plt.show() # 展示
# plt.savefig('xxx.png') # 保存图标函数

fontsize用于设置各轴的字体大小。

绘制单点图

scatter函数绘制单点图

import matplotlib.pyplot as plt

plt.scatter(1,8,s=50,c='red') 
plt.show()

参数s设置了绘制图形时使用的点的尺寸。c参数设置点的颜色，也可以通过RDB方式来指定：c=(0,0,0.8)

绘制一系列点

import matplotlib.pyplot as plt
x_values = list(range(1001))
y_values = [x**2 for x in x_values]
plt.scatter(x_values,y_values,s=40,edgecolors='none') # edgecolors参数用于去除外轮廓
plt.show()

edgecolors可以指定white或者black等。

绘制颜色映射图

import matplotlib.pyplot as plt


x_values = list(range(1001))
y_values = [x**2 for x in x_values]
plt.scatter(x_values,y_values,s=40,edgecolors='none',c=y_values,cmap=plt.cm.Blues)

将y_values参数传入c中，并使用cmap告诉函数使用哪个颜色进行映射。