如何绘制从Python中返回多个值的函数的输出?

时间:2016-02-18 16:19:38

标签: python matplotlib plot physics multipleoutputs

简介

我计算和绘制来自给定数据的行星轨道脉冲星的光谱能量。

我之前订购了带有尺寸[172,2](172行和2列)的列表变种中的所有数据。

首先,我必须相应地计算预设模型和光谱能量的参数(来自那些参数)。

为此,我定义了一个函数,在该函数下我定义了预先设定的模型和 find_fit 函数,该函数获取模型和变体数据。

代码

 var('a, b, t')

def spectrum(omega):

    model = a*sin(omega*t) + b*cos(omega*t)
    fit = find_fit(variations, model, parameters= [a, b], variables = [t], solution_dict = True)
    sp_en = ((fit[a])**2 + (fit[b])**2)/2

    return fit[a], fit[b], sp_en

然后我调用函数并打印值:

c, v, energy = spectrum(20)   #enter arbitray angular frequency here
print "Values for the given angular frequency : \n \n a = %f, b = %f, spectral_energy = %f " % (c, v, energy)

现在我必须绘制 sp_en 输出。

"半溶液"

如果频谱函数仅返回 sp_en ,则很容易。这足以写:

var('t')
plot(spectrum(t), (t, 1, 100))

返回: energy-omega plot

问题不在于:如果我想打印所有三个输出,如何绘制此功能?

2 个答案:

答案 0 :(得分:0)

只使用energy变量

调用绘图函数
omega=10
c, v, energy = spectrum(omega)   #enter arbitray angular frequency here
print "Values for the given angular frequency : \n \n a = %f, b = %f, spectral_energy = %f " % (c, v, energy)

plot(energy, (omega, 1, 100)

答案 1 :(得分:0)

只需对频谱的返回值使用索引:

plot(spectrum(t)[2], (t, 1, 100))