matplotlib画图实例

记录下，matplotlib画图的例子，方便后续查询：

#-*- encoding=utf-8 -*-
import matplotlib.pyplot as plt
import numpy as np

# make data
x = [1, 1, 1]

colors = plt.get_cmap('Blues')(np.linspace(0.2, 0.7, len(x)))

# plot
fig, ax = plt.subplots(2, 2, figsize=(8, 8))
# plt.figure(figsize=(8, 8))

ax[0][0].pie(x, colors=colors, radius=3, center=(4, 4),
       wedgeprops={"linewidth": 1, "edgecolor": "white"}, frame=True)

ax[0][0].set(xlim=(0, 8), xticks=np.arange(1, 8),
       ylim=(0, 8), yticks=np.arange(1, 8))

an = np.linspace(-2*np.pi, 2*np.pi, 100)
bn = np.round(np.linspace(-2*np.pi, 2*np.pi, 10),1)

ax[0][1].plot(an, np.sin(an))
ax[0][1].plot(an, np.cos(an))
ax[0][1].plot(an, np.cos(an + np.pi/2.0))

ax[0][1].set(xlim=(-np.pi*2, np.pi*2), xticks=np.linspace(-np.pi*2,np.pi*2,10),
       ylim=(-1, 1), yticks=np.linspace(-1,1,10))

ax[0][1].set_xlabel("asd", rotation=90)

ax[0][1].set_xticks(bn)
ax[0][1].set_xticklabels(bn, rotation=90)

ax[1][0].plot(an, np.cos(an))

ax[1][1].plot(an, np.tan(an))

plt.show()