matplotlib坐标投影

LeonDL168

于 2024-09-18 09:27:19 发布

阅读量6

点赞数

分类专栏： Python 文章标签：算法人工智能视觉检测图像处理计算机视觉

原文链接：https://tinycool.blog.csdn.net/article/details/136243868

版权

Python 专栏收录该内容

4 篇文章 0 订阅

订阅专栏

文章目录

matplotlib教程：初步📈 子图绘制

三维坐标

【plt】可以将二维图形嵌入到三维坐标系中，方法也很简单，只需用创建一个子坐标框架，并指定其投影方式为3d，并通过zdir参数指定一个额外的轴即可。

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$ 曲线，但plot函数是有实力绘制真正的三维曲线的。但在三维空间中，需要三个坐标才能确定一个点的位置，空间中的一条三维曲线，需要三个坐标来共同表示

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设为polar即可实现极坐标的绘制。并且在极坐标中，plot函数保留了默认自变量的功能，示例如下

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是一个很大的数，相当于三分之一个圆周了，所以在右侧的极坐标图像中，出现了明显的棱角。

地理坐标

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