「Python」matplotlib备忘录

总结了一下网上现有的资源，得到了一些东西。随手做个备忘。

更多设置见：https://matplotlib.org/users/customizing.html。

导入

import matplotlib.pyplot as plt

from pylab import mpl

mpl.rcParams['font.sans-serif'] = ['Noto Sans SC'] # 避免中文在Linux下出现方框

mpl.rcParams['axes.linewidth'] = 1.5 # 设置坐标轴线宽

import seaborn as sns # 比较好看的绘图风格

sns.set() # 设置

在PyCharm中使用

%matplotlib inline

%config InlineBackend.figure_format = 'svg'

# 保存图片

plt.savefig("pict.png", bbox_inches='tight')

plt.show()

绘图

基础绘图

plt.figure(figsize=(8,6), dpi=120)

plt.subplot(1,1,1) # 创建一个1x1的子图，接下来的图片绘制在第1块

X = np.linspace(-np.pi, np.pi, 256,endpoint=True) # 范围是[-pi,pi]，个数是250个。如果不加endpoint属性是(-pi,pi)

C,S = np.cos(X), np.sin(X)

# 绘制余弦曲线，使用蓝色的、连续的、宽度为 1 （像素）的线条

plt.plot(X, C, color="blue", linewidth=1.0, linestyle="-")

# 绘制正弦曲线，使用绿色的、连续的、宽度为 1 （像素）的线条

plt.plot(X, S, color="green", linewidth=1.0, linestyle="-")

# 设置横轴的上下限

plt.xlim(-4.0,4.0)

# 设置横轴记号

plt.xticks(np.linspace(-4,4,9,endpoint=True))

# 设置纵轴的上下限

plt.ylim(-1.0,1.0)

# 设置纵轴记号

plt.yticks(np.linspace(-1,1,5,endpoint=True))

# 增加图例，在右上角

plt.legend(loc='upper right')

# 以分辨率 72 来保存图片

# plt.savefig("exercice_2.png",dpi=72)

# 在屏幕上显示

plt.show()

三维绘图

from mpl_toolkits.mplot3d import Axes3D

import numpy as np

import matplotlib.pyplot as plt

fig = plt.figure()

ax = Axes3D(fig)

x=np.arange(-2*np.pi,2*np.pi,0.1)

y=np.arange(-2*np.pi,2*np.pi,0.1)

X, Y = np.meshgrid(x, y) #网格的创建，这个是关键

Z=np.sin(X)*np.cos(Y)

plt.xlabel('x')

plt.ylabel('y')

ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap='rainbow')

plt.show()

最终图像：