お世話になっております。
Pythonを使用して、極大値を算出したいです。
数値が記入されているExcelファイルにargrelmaxを使用して求めようと思いましたが、うまくいきません。
独学でプログラミングを始めたばかりですので、初歩的な質問で申し訳ございません。
やりたいこととしては、一つのグラフより、グラフの立ち上がりと立下りよりそれぞれを抜き出して保存していきたいです。
df1=df["water_lev"] print(df1) 0 -2.09 1 -2.09 2 -2.09 3 -2.08 4 -2.08 ... 8779 -2.33 8780 -2.35 8781 -2.37 8782 -2.39 8783 -2.40 Name: water_lev, Length: 8784, dtype: float64
import numpy as np from scipy import signal from scipy import fftpack from matplotlib import pyplot as plt df = pd.read_excel('test.xlsx') df1=df["water_lev"] # フーリエ変換して複素スペクトル、振幅、周波数を出力する関数 def fft(df1, dt): spec = fftpack.fft(df1) # フーリエスペクトル(複素数) freq = fftpack.fftfreq(len(spec), d=dt) # 周波数軸を生成 amp = np.sqrt(spec.real ** 2 + spec.imag ** 2) / (len(spec)/2) # 振幅を計算 return spec, amp, freq # 波形(x, y)からn個のピークを幅wで検出する関数(xは0から始まる仕様) def findpeaks(df1, n, w): index_all = list(signal.argrelmax(df1, order=w)) # scipyのピーク検出 index = [] # ピーク指標の空リスト peaks = [] # ピーク値の空リスト # n個分のピーク情報(指標、値)を格納 for i in range(n): index.append(index_all[0][i]) peaks.append(y[index_all[0][i]]) index = np.array(index) * x[1] # xの分解能x[1]をかけて指標を物理軸に変換 return index, peaks # フーリエ変換する spec, amp, freq = fft(df1, dt) # ピーク検出する index, peaks = findpeaks(freq, amp, n=100, w=6)
あなたの回答
tips
プレビュー