scipy 对齐多条曲线的峰值

imzjd6km 于 2022-11-23 发布在其他

关注(0)|答案(1)|浏览(245)

我有一个带有x和y值的数据集。曲线被绘制出来，峰值位于x轴沿着的不同值处。
我试图用scipy的信号来对齐所有曲线的峰值。我试着跟随这个帖子Use of pandas.shift() to align datasets based on scipy.signal.correlate，但是峰值不重叠。

import matplotlib.pyplot as plt
from scipy import signal
import math
import numpy as np

a = [0.0002, 0.0005, 0.009, 0.0207, 0.0307, 0.04, 0.044, 0.05, 0.07, 0.07, 0.07, 0.082, 0.087, 0.089, 0.09, 0.09, 0.097,
     0.1, 0.11, 0.149, 0.153, 0.159, 0.16, 0.16, 0.2, 0.24, 0.24, 0.24, 0.25, 0.27, 0.3, 0.385, 0.46, 0.77, 3.7]
b = [0.4, 0.48, 2.2, 2.2, 3.4, 4.0, 4.7, 7.15, 9.9]
c = [0.006, 0.01, 0.01, 0.01, 0.012, 0.013, 0.0178, 0.018, 0.02, 0.022, 0.022, 0.027, 0.031, 0.035, 0.035, 0.036, 0.04,
     0.04, 0.046, 0.046, 0.047, 0.05, 0.0507, 0.06, 0.062, 0.07, 0.071, 0.08, 0.1, 0.143, 0.18, 0.19, 0.255, 0.3, 0.4,
     0.75, 1.25, 4.8, 35.0, 100.0]
d = [0.002, 0.01, 0.012, 0.018, 0.032, 0.035, 0.042, 0.13, 0.14, 0.172]
e = [0.0033, 0.01, 0.012, 0.023, 0.023]

data = {'a': a, 'b': b, 'c': c, 'd': d, 'e': e}

fig = plt.figure()
xc = [*range(0, len(data['c']), 1)]
for k, v in data.items():

    x = [*range(0, len(data[k]), 1)]
    v = [math.log10(i) for i in v]
    # https://stackoverflow.com/questions/10482684/python-reorder-a-sorted-list-so-the-highest-value-is-in-the-middle
    v = v[len(v) % 2::2] + v[::-2]
    # plt.plot(x, [math.log10(i) for i in v], '*')
    if k == 'c':
        plt.plot(xc, v, '*', linestyle='--')
    dx = np.mean(np.diff(xc))
    shift = (np.argmax(signal.correlate(data['c'], v)) - len(v)) * dx
    if k != 'c':
        plt.plot(x + shift, v)

峰值并不以相同的x值为中心。关于如何做到这一点的建议将非常有帮助。

scipy

来源：https://stackoverflow.com/questions/74500739/align-the-peaks-of-multiple-curves

1条答案

按热度按时间

zed5wv101#

import matplotlib.pyplot as plt
import numpy as np

def centralize(v):
    return np.concatenate((v[len(v) % 2::2], v[::-2]))

def shift_x(v, max_position):
    
    start = max_position - v.argmax()
    end = start + len(v)
    x = np.arange(start, end)
    return x

a = [0.0002, 0.0005, 0.009, 0.0207, 0.0307, 0.04, 0.044, 0.05, 0.07, 0.07, 0.07, 0.082, 0.087, 0.089, 0.09, 0.09, 0.097,
     0.1, 0.11, 0.149, 0.153, 0.159, 0.16, 0.16, 0.2, 0.24, 0.24, 0.24, 0.25, 0.27, 0.3, 0.385, 0.46, 0.77, 3.7]

b = [0.4, 0.48, 2.2, 2.2, 3.4, 4.0, 4.7, 7.15, 9.9]

c = [0.006, 0.01, 0.01, 0.01, 0.012, 0.013, 0.0178, 0.018, 0.02, 0.022, 0.022, 0.027, 0.031, 0.035, 0.035, 0.036, 0.04,
     0.04, 0.046, 0.046, 0.047, 0.05, 0.0507, 0.06, 0.062, 0.07, 0.071, 0.08, 0.1, 0.143, 0.18, 0.19, 0.255, 0.3, 0.4,
     0.75, 1.25, 4.8, 35.0, 100.0]

d = [0.002, 0.01, 0.012, 0.018, 0.032, 0.035, 0.042, 0.13, 0.14, 0.172]

e = [0.0033, 0.01, 0.012, 0.023, 0.023]

data = {'a': a, 'b': b, 'c': c, 'd': d, 'e': e}

fig = plt.figure()

xc = np.arange(len(c))
vc = centralize(np.log10(c))
max_position = vc.argmax()

plt.plot(xc, vc, '*', linestyle='--')
for k, v in data.items():

    if k != 'c':
        v = centralize(np.log10(v))
        x = shift_x(v, max_position)
        plt.plot(x, v)

plt.show()

赞(0）回复(0）举报 2022-11-23

我来回答

scipy 对齐多条曲线的峰值

1条答案

相关问题

热门标签

最新问答