首页  

Python识别波峰波谷     所属分类 quant 浏览量 167
import numpy as np

data = [3, 5, 2, 8, 1, 9, 4, 7]
data_array = np.array(data)

from scipy.signal import find_peaks

# 计算波峰
# 注意 用 python3 !!!
peaks, _ = find_peaks(data_array)

# 索引值 array([1, 3, 5])
# peaks

# 计算波谷
# Numpy数组前加负号(-) ,表示取数组的逆序
valleys, _ = find_peaks(-data_array)
#  索引值 [2 4 6]
# valleys

import matplotlib.pyplot as plt

# 绘制原始数据
plt.plot(data_array)
# 标注波峰和波谷
plt.plot(peaks, data_array[peaks], "x", color='red')
plt.plot(valleys, data_array[valleys], "x", color='blue')
plt.show()

上一篇     下一篇
vue3创建项目并运行

vue3 todolist项目 运行

智能搜索工具

talib简介

程序员 如何利用 个人网站 自媒体赚钱

springboot3 自动配置文件