垂直填充3D Matplotlib图表

Question

我有一个使用matplotlib制作的3D图。现在我想填充绘制线条和x，y轴之间的垂直空间，以突出显示该线在z轴上的高度。在2D图中，可以使用“fill_between”完成此操作，但似乎没有类似于3D图的功能。有人能帮助吗？

这是我目前的代码：

from stravalib import Client
import matplotlib as mpl
import numpy as np
import matplotlib.pyplot as plt

... code to get the data ....

mpl.rcParams['legend.fontsize'] = 10

fig = plt.figure()
ax = fig.gca(projection='3d')
zi = alt
x = df['x'].tolist()
y = df['y'].tolist()
ax.plot(x, y, zi, label='line')
ax.legend()
plt.show()     

当前的情节是什么？

为了更清楚，我想要一个竖直填充到x，y轴交点，而不是这个...

- EnduroDave

Answer 1

你是对的。似乎在3D图中没有与2D图函数fill_between相当的东西。我建议的解决方案是将您的数据转换为3D多边形。以下是相应的代码：

import math as mt
import matplotlib.pyplot as pl
import numpy as np
import random as rd

from mpl_toolkits.mplot3d import Axes3D
from mpl_toolkits.mplot3d.art3d import Poly3DCollection


# Parameter (reference height)
h = 0.0

# Code to generate the data
n = 200
alpha = 0.75 * mt.pi
theta = [alpha + 2.0 * mt.pi * (float(k) / float(n)) for k in range(0, n + 1)]
xs = [1.0 * mt.cos(k) for k in theta]
ys = [1.0 * mt.sin(k) for k in theta]
zs = [abs(k - alpha - mt.pi) * rd.random() for k in theta]

# Code to convert data in 3D polygons
v = []
for k in range(0, len(xs) - 1):
    x = [xs[k], xs[k+1], xs[k+1], xs[k]]
    y = [ys[k], ys[k+1], ys[k+1], ys[k]]
    z = [zs[k], zs[k+1],       h,     h]
    #list is necessary in python 3/remove for python 2
    v.append(list(zip(x, y, z))) 
poly3dCollection = Poly3DCollection(v)

# Code to plot the 3D polygons
fig = pl.figure()
ax = Axes3D(fig)
ax.add_collection3d(poly3dCollection)
ax.set_xlim([min(xs), max(xs)])
ax.set_ylim([min(ys), max(ys)])
ax.set_zlim([min(zs), max(zs)])
ax.set_xlabel("x")
ax.set_ylabel("y")
ax.set_zlabel("z")

pl.show()

它会生成以下图形： 希望这能帮到您。

- Flabetvibes

---

对于Python 3，您需要将v.append(zip(x,y,z))更改为v.append(list(zip(x,y,z)))。 - Ianhi