Как создать 3D-график, используя функцию, которая возвращает график? - PullRequest
Новая
       28

Как создать 3D-график, используя функцию, которая возвращает график?

0 голосов
/ 05 июня 2018

Я хотел бы создать трехмерный график, используя функцию, которая возвращает график и входной аргумент, который он принимает.Это мой код функции: 

def cumulative(moment):
    bins = np.zeros(32)
    x = upper_bin
    for i in range(32):
        bins[i] = burst_average[moment, 0:i+1].sum()
    plt.ylim(ymax = 1000)
    plt.xlabel('grain size (um)')
    plt.ylabel('concentration (uL/L)')
    plt.title('grain size distribution over time')
    plt.plot(x, bins, c = 'b', label=dates[i])
    return 

import ipywidgets as widgets
from ipywidgets import interact

interact(cumulative, moment=widgets.FloatSlider(min = int(0), max = int(nr_burst-1), step = 1, description = 'moment'));

, где x - это список из 32 значений, bins - это также массив из 32 значений, который изменяется для каждого moment.В общей сложности создаются графики nr_burst, что составляет около 2017 года. Виджет работает, однако я хочу включить это в свой отчет, поэтому вместо этого я хотел бы использовать трехмерный график.

Я пытался что-то вроде 

from mpl_toolkits.mplot3d import Axes3D
from mpl_toolkits import mplot3d    

b0 = np.linspace(0, nr_burst-1, nr_burst)
b= []
for i in range(len(b0)):
    b.append(int(b0[i]))

ax.scatter3D(cumulative(b), b)

Это не сработало, выдал ошибку ValueError: Arguments 'xs' and 'ys' must be of same size.

Я также попытался функцию, чтобы вернуть x и bи plot как 

ax.scatter3D(cumulative(b)[0], b, cumulative(b)[1])

, который дал ошибку TypeError: 'NoneType' object is not subscriptable.

1 Ответ

0 голосов
/ 05 июня 2018

После построения исходных данных используйте: 

ax = plt.gca() # get the axis handle of the current graphic artist

data_2d = ax.lines[0] # this just extracts the first dataset

x,y = data_2d.get_xdata(), data_2d.get_ydata() #this will be your x and y data

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

ax.scatter3D(x, b, y)

Второй вариант

Измените исходную функцию, чтобы она возвращала дескриптор оси. 

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

def cumulative(moment):
    fig, ax = plt.subplots()
    bins = np.cumsum(np.arange(moment))
    x = np.arange(moment)
    ax.plot(x, bins, c = 'b')
    ax.set_xlabel('grain size (um)')
    ax.set_ylabel('concentration (uL/L)')
    ax.set_title('grain size distribution over time')
    ax.set_ylim(ymax = bins.max())
    return fig, ax

b = 32 #just a random scalar to test

fig, ax = cumulative(b) #call the function and assign the returning values

data_2d = ax.lines[0] # get your data
x,y = data_2d.get_xdata(), data_2d.get_ydata() #your data separated for x and y

plot3d = plt.figure()
ax3d = plot3d.add_subplot(111, projection='3d')
ax3d.scatter(x,b,y)
...