在下面的代码中,我有两个高斯曲线,一个是红色,另一个是紫色曲线。我想知道python中是否有一种方法可以将两个高斯曲线组合到第三条曲线上,这条曲线看起来像蓝色曲线(它只是作为高斯曲线的一个例子,据说更高更宽)?任何帮助将不胜感激。
import numpy as np
import scipy.optimize as opt
import matplotlib.pyplot as plt
def gauss(x, p): # p[0]==mean, p[1]==stdev, p[2]==heightg, p[3]==baseline
a = p[2]
mu = p[0]
sig = p[1]
base = p[3]
return a * np.exp(-1.0 * ((x - mu)**2.0) / (2.0 * sig**2.0)) + base
p0 = [6804.5, 1.2, 23.0, 25.3532] # Inital guess is a normal distribution
p02 = [6804.5, 6.5, 5.0, 25.09098]
xp = np.linspace(6780, 6810, 200)
fig = plt.figure()
a1 = fig.add_subplot(111)
a1.plot(xp, gauss(xp, p0), lw=3, alpha=2.5, color='r')
a1.plot(xp, gauss(xp, p02), lw=3, alpha=2.5, color='purple')
a1.set_xlim([6798, 6810])
plt.tight_layout()
plt.show()