将Python 2D图转为3D(Matplotlib)

4 投票
1 回答
5443 浏览
提问于 2025-04-17 22:21

在Python的Matplotlib库中绘图:我每天在同一时间采集了一些样本,这些样本显示了某种测量值的变化。这个变化可以用一个二维图来展示(左边的图),但是随着样本数量的增加,我想把这些数据展示成一个三维的堆叠图(右边的图)——这个图只是用来说明的。

作为一个起点,我的代码如下,我该如何实现这个目标呢?

import numpy as np
import pylab as plt


t  = np.arange(1024)*1e-6
y1 = np.sin(t*2e3*np.pi) 
y2 = 0.5*y1
y3 = 0.25*y1

plt.plot(t,y1,'k-', label='12/03/14')
plt.plot(t,y2,'r-', label='13/03/14')
plt.plot(t,y3,'b-', label='14/03/14')
plt.xlabel('Time/sample no.')
plt.ylabel('Pk-pk level (arbitrary units)')
plt.legend()
plt.grid()
plt.show()

enter image description here

1 个回答

5

这大概是这样的吧?

from mpl_toolkits.mplot3d import Axes3D
from matplotlib.collections import PolyCollection
from matplotlib.colors import colorConverter
import matplotlib.pyplot as plt
import numpy as np

fig = plt.figure()
ax = fig.gca(projection='3d')

zs = [0.0, 1.0, 2.0]
t  = np.arange(1024)*1e-6
ones = np.ones(1024)
y1 = np.sin(t*2e3*np.pi) 
y2 = 0.5*y1
y3 = 0.25*y1

verts=[list(zip(t, y1)), list(zip(t, y2)), list(zip(t, y3))]


poly = PolyCollection(verts, facecolors = ['r','g','b'])
poly.set_alpha(0.7)
ax.add_collection3d(poly, zs=zs, zdir='y')
ax.set_xlabel('X')
ax.set_xlim3d(0, 1024e-6)
ax.set_ylabel('Y')
ax.set_ylim3d(-1, 3)
ax.set_zlabel('Z')
ax.set_zlim3d(-1, 1)

plt.show()

在这里输入图片描述

撰写回答