pandas - 在 pandas 和 scipy.signal 中查找局部最大值/最小值的索引
问题描述
我的目标是在 pandas 或 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()
所以,我不知道如何从这一点继续并在图上找到与获得的局部最大值和最小值相对应的索引。将不胜感激任何帮助!
解决方案
我你使用df['min'].notna()
它返回一个数据框,其中的行min
不是nan
。要查找局部最小值的索引,您可以使用该.loc
方法。
df.loc[df['min'].notna()].index
您的示例的输出结果是:
Int64Index([0, 11, 21, 35, 43, 54, 67, 81, 105, 127, 141, 161, 168, 187], dtype='int64')
您可以使用类似的过程来查找局部最大值。
推荐阅读
- python - 随机打乱numpy数组每行中的项目
- mysql - 表格行或列
- php - 在增加 php bootstrap 下拉菜单后恢复 foreachloop
- javascript - 在jQuery中将一个变量从一个函数传递给另一个函数
- java - JAVA 从文件中拆分文本并转换为消息
- google-apps-script - 如何在 getRange 中使用 var
- javascript - 事件侦听器回调和 MutationObserver 回调的执行顺序是否准确?
- python - 我如何获得返回码?
- r - 在工具提示中显示额外的变量 echarts4r
- amazon-web-services - 如何检测超过 50 个单词的图像中的 TEXT?