matplotlib画三维曲面Demo

最新推荐文章于 2024-09-18 11:03:32 发布

写代码的资资

最新推荐文章于 2024-09-18 11:03:32 发布

阅读量485

点赞数

分类专栏： Python

本文链接：https://blog.csdn.net/weixin_43118073/article/details/113432344

版权

Python 专栏收录该内容

79 篇文章 4 订阅

订阅专栏

这个UP写的很好！

需要修补，由 ax.plot_trisurf 函数完成。
首先找到一组所有点都连接起来的三角形，然后用这些三角形创建曲面
在这里插入图片描述

from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
from matplotlib import cm
import numpy as np
from sys import argv

#x,y,z = np.loadtxt('your_file', unpack=True)
x = np.random.rand(500)*10
y = np.random.rand(500)*10
z = np.random.rand(500)*10
fig = plt.figure()
ax = Axes3D(fig)
surf = ax.plot_trisurf(x, y, z, cmap=cm.rainbow, linewidth=0.1)
fig.colorbar(surf, shrink=0.5, aspect=5)
plt.savefig('teste.png',dpi=300)
plt.show()

在这里插入图片描述

import matplotlib.pyplot as plt
import numpy as np

from matplotlib import cm
from matplotlib.ticker import LinearLocator,FormatStrFormatter
from mpl_toolkits.mplot3d import Axes3D

fig = plt.figure()
ax = fig.add_subplot(1,1,1,projection="3d")

x = np.arange(-5, 5, 0.25)
y = np.arange(-5, 5, 0.25)
x, y = np.meshgrid(x, y)
R = np.sqrt(x**2 + y**2)
z = np.sin(R)
# x = np.arange(-3,3,0.25)
# y = np.arange(-3,3,0.25)
# x,y = np.meshgrid(x,y)
# r = np.sqrt(np.power(x,2)+np.power(y,2))
# z = np.sin(r)

# plot 3d surface
surf = ax.plot_surface(x,y,z,
                       rstride=1,
                       cstride=1,
                       cmap=cm.coolwarm,
                       linewidth=0,
                       antialiased=False)

# customize the z axis
ax.set(zlim=(-1,1))
ax.zaxis.set_major_locator(LinearLocator(7))
ax.zaxis.set_major_formatter(FormatStrFormatter("%3.2f"))

# add a color bar mapping values to colors
fig.colorbar(surf,shrink=0.6,aspect=10)
ax.xaxis.set_pane_color((1.0, 1.0, 1.0, 0.0))
ax.yaxis.set_pane_color((1.0, 1.0, 1.0, 0.0))
ax.zaxis.set_pane_color((1.0, 1.0, 1.0, 0.0))
ax.grid(linewidth=20,color=None)
plt.savefig('3D_surf.png',dpi=300)
plt.show()