跟踪数组中索引中的Y

时间:2019-02-18 01:00:01

标签: python numpy

我似乎无法更新元组中的Y值。 这是我的功能应该执行的操作: 创建上下浮动的数据数组。

def make_jump(base_val,start_width,peak_array):

    num_samps = start_width + np.sum([x[:2] for x in peak_array])

    signal = np.zeros(int(num_samps))

    x0=start_width
    signal[:start_width] = base_val*np.ones(start_width)

    for idx, row in enumerate(peak_array):

        samps_to_peak = int(x0+row[0])

        signal[start_width:samps_to_peak+1] = np.linspace(base_val,base_val+row[2], len(signal[start_width:samps_to_peak+1])) 

        peak_width = int(samps_to_peak+1+row[1])

        peak_height = int(peak_width + row[2])
        signal[samps_to_peak+1:peak_width] = np.linspace(base_val+row[2],base_val+row[2],len(signal[samps_to_peak+1:peak_width]))

    return signal

peak_tuple = np.array([[2, 5, 45],
                       [3,7,20]])

plt.plot(make_jump(base_val=0, start_width = 5, peak_array = peak_tuple))

目标是制作两个峰,第一个峰为45,在5个样本中保持在45,然后在两个样本中从0到45的基值到达该峰。之后,应该取三个样本达到65的高度,以获取四个样本。如何跟踪每个峰之后的y值,以及保存到下一个峰的y值?如果我们取出[3,7,20],它就会按照预期的方式工作。

0 个答案:

没有答案