python matplotlib实现坐标投影的示例代码

import numpy as np
import matplotlib.pyplot as plt
xs = np.linspace(0, 1, 100)
ys = np.sin(xs * 2 * np.pi) / 2 + 0.5

fig = plt.figure()
zs = ['x', 'y', 'z']
for i, z in enumerate(zs, 1):
    ax = fig.add_subplot(1,3,i, projection='3d')
    ax.plot(xs, ys, zs=0, zdir=z)
    ax.set_zlim(0,1)
    ax.set_xlim(0,1)
    ax.set_ylim(0,1)

plt.show()

xs = np.linspace(0, 5, 100)
ys = np.sin(xs * 2 * np.pi) / 2 + 0.5
zs = np.cos(xs * 2 * np.pi) / 2 + 0.5
ax = plt.subplot(projection='3d')
ax.plot(xs, ys, zs)
plt.show()

r = np.arange(0, 2, 0.01)
th = 2 * np.pi * r

ax = plt.subplot(121, projection='polar')
ax.plot(th, r)
ax = plt.subplot(122, projection='polar')
ax.plot(r)
plt.show()

projs = ['aitoff', 'hammer', 'lambert', 'mollweide']
fig = plt.figure()
xs = ys = np.arange(-2,2,0.01)
for i, p in enumerate(projs, 1):
    ax = fig.add_subplot(2,2,i,projection=p)
    ax.plot(xs, ys)
    plt.title(p)
    plt.grid()

plt.show()