Как построить функцию, ориентированную на локальную ось X Matplotlib 3D? - PullRequest
0 голосов
/ 14 сентября 2018

Мне нужно нанести функцию af (z) на 3D matplotlib, но на локальной оси x, определяемой двумя точками и заполненной между ними, чтобы остаться, как показано на рисунке:

here

У меня есть эти две точки, которые определяют локальную ось x, 20 значений между ними и соответствующие 20 значений f (z), но я не могу понять, как построить график.Кто-нибудь может мне помочь?

valuesx = np.arange(0.0, 5, 5/20) # local axis values ​​x
self.listax.append(valuesx)                   
for l in valuesx:
    fy= 2*x**2-4 #equation
    fyx = eval(fy, {'x': l})
    self.listay.append(fyx)
x = [self.listax]
y = [self.listay]
z = [1, 5] 
verts = [list(zip(x, y, z))]               
self.axes.add_collection3d(Poly3DCollection(verts, facecolor = 'red', alpha=0.6), zs='z')
self.fig.canvas.draw() 

1 Ответ

0 голосов
/ 14 сентября 2018

Извините, это немного быстрый и грязный ответ, но вам может помочь следующий пример:

https://matplotlib.org/gallery/mplot3d/polys3d.html

Адаптация приведенного выше к вашему примеру:

from mpl_toolkits.mplot3d import Axes3D
from matplotlib.collections import PolyCollection
import matplotlib.pyplot as plt
from matplotlib import colors as mcolors
import numpy as np
%matplotlib inline

def f(x):
    return (2*x**2-4)

valuesx = np.arange(0.0, 5, 5/20) 
valuesy= np.array([f(i) for i in valuesx])

def polygon_under_graph(xlist, ylist):
    '''
    Construct the vertex list which defines the polygon filling the space under
    the (xlist, ylist) line graph.  Assumes the xs are in ascending order.
    '''
    return [(xlist[0], 0.)] + list(zip(xlist, ylist)) + [(xlist[-1], 0.)]

zs = 0

fig = plt.figure()
ax = fig.gca(projection='3d')
verts=[]
verts.append(polygon_under_graph(valuesx, valuesy))

poly = PolyCollection(verts, facecolors='r')
ax.add_collection3d(poly, zs=zs, zdir='x')

ax.set_xlim(0, 5)
ax.set_ylim(0, 4)
ax.set_zlim(np.min(ys), np.max(ys))

Должно дать вам:

enter image description here

Затем вы можете отрегулировать пределы по своему усмотрению и отрегулировать переменную zs для отображения вдоль значения наось х.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...