我的目标是在panda或matplotlib中找到函数的局部最大值和最小值的指数。
假设我们有一个噪声信号,其局部最大值和最小值已经绘制,如以下链接所示:
https://stackoverflow.com/a/50836425/15934571
下面是代码(我只是从上面的链接复制粘贴):
import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
from scipy.signal import argrelextrema
# Generate a noisy AR(1) sample
np.random.seed(0)
rs = np.random.randn(200)
xs = [0]
for r in rs:
xs.append(xs[-1] * 0.9 + r)
df = pd.DataFrame(xs, columns=['data'])
n = 5 # number of points to be checked before and after
# Find local peaks
df['min'] = df.iloc[argrelextrema(df.data.values, np.less_equal,
order=n)[0]]['data']
df['max'] = df.iloc[argrelextrema(df.data.values, np.greater_equal,
order=n)[0]]['data']
# Plot results
plt.scatter(df.index, df['min'], c='r')
plt.scatter(df.index, df['max'], c='g')
plt.plot(df.index, df['data'])
plt.show()
所以,我不知道如何继续从这一点,并找到相应的指数获得局部最大值和最小值的曲线。将不胜感激任何帮助!
1条答案
按热度按时间f87krz0w1#
使用
df['min'].notna()
,它返回一个 Dataframe ,其中min
的行不是nan
。要查找局部最小值的索引,可以使用.loc
方法。示例的输出结果为:
你可以用类似的过程求局部极大值。