python matplotlib實現(xiàn)坐標投影的示例代碼

更新時間：2024年02月26日 11:56:40 作者：微小冷

這篇文章主要為大家詳細介紹了python matplotlib實現(xiàn)坐標投影,文中的示例代碼講解詳細,具有一定的參考價值,感興趣的小伙伴可以跟隨小編一起學習一下

三維坐標

【plt】可以將二維圖形嵌入到三維坐標系中，方法也很簡單，只需用創(chuàng)建一個子坐標框架，并指定其投影方式為3d，并通過zdir參數(shù)指定一個額外的軸即可。

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()

在這里插入圖片描述

很顯然，這是個偽 3 D 3D 3D曲線，但plot函數(shù)是有實力繪制真正的三維曲線的。但在三維空間中，需要三個坐標才能確定一個點的位置，空間中的一條三維曲線，需要三個坐標來共同表示

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()

效果如下

在這里插入圖片描述

極坐標

極坐標也是一種投影，只需將projection設(shè)為polar即可實現(xiàn)極坐標的繪制。并且在極坐標中，plot函數(shù)保留了默認自變量的功能，示例如下

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()

在這里插入圖片描述

對于弧度而言，1是一個很大的數(shù)，相當于三分之一個圓周了，所以在右側(cè)的極坐標圖像中，出現(xiàn)了明顯的棱角。

地理坐標

projection的含義是投影，除了3D坐標和極坐標之外，還支持多種地理坐標，下面就列舉幾個不同的地理坐標并繪圖

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()

在這里插入圖片描述