Scipyのargrelmaxを使って極値を算出し、再度リストに入れたいが上手くいかない。

Question

前提・実現したいこと

python3.6のScipyライブラリを使って、あるデータの極値を求めようと考えています。
argrelmax()、argrelmin()という関数を使えば、Numpy配列から作ったグラフの極値のインデックスを取得できることを知ったのですが、そこから極値に該当するデータのみをNumpy配列やリストに収める方法がわからず、頭を抱えています。
ソースコードの
e_center.txt
には、改行で区切られたグラフの値(x-y座標でいうところのyの値のみ)が入っています。

発生している問題・エラーメッセージ

C:\Users\USER\python\FDTD\シン・ELECTROMAGNETIC SIMULATION USING THE FDTD(202201
10より)\テストコード>python for_envelope.py
Traceback (most recent call last):
  File "for_envelope.py", line 22, in <module>
    maxid = signal.argrelmax(e_center_raw_n, order=1)
  File "C:\Users\USER\AppData\Local\Programs\Python\Python36\lib\site-packages\s
cipy\signal\_peak_finding.py", line 191, in argrelmax
    return argrelextrema(data, np.greater, axis, order, mode)
  File "C:\Users\USER\AppData\Local\Programs\Python\Python36\lib\site-packages\s
cipy\signal\_peak_finding.py", line 246, in argrelextrema
    axis, order, mode)
  File "C:\Users\USER\AppData\Local\Programs\Python\Python36\lib\site-packages\s
cipy\signal\_peak_finding.py", line 74, in _boolrelextrema
    results &= comparator(main, plus)
numpy.core._exceptions.UFuncTypeError: ufunc 'greater' did not contain a loop wi
th signature matching types (dtype('<U23'), dtype('<U23')) -> dtype('bool')

該当のソースコード

#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""
包絡線を書く用
"""

import  numpy  as  np 
import scipy.constants as sc
from scipy import signal


with open('e_center.txt') as f:
    e_center_raw = [s.strip() for s in f.readlines()]

e_center_raw_n = np.array(e_center_raw)

maxid = signal.argrelmax(e_center_raw_n, order=1) 
minid = signal.argrelmin(e_center_raw_n, order=1) 

e_center_env = e_center_raw_n[maxid]

f = open('e_center_envelope.txt', 'w')
for x in e_center_env:
    f.write(str(x) + "\n")
f.close()

補足情報（FW/ツールのバージョンなど）

scipy 1.5.4
numpy 1.19.5

Answer 1

データの中身がわかりませんが、'12345678901234567890123'のようなものが入っているとそうなります。

>>> e_center_raw = ['12345678901234567890123', '12345678901234567890125', '12345678901234567890120']
>>> e_center_raw_n = np.array(e_center_raw)
>>> maxid = signal.argrelmax(e_center_raw_n, order=1)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "C:\Users\shinp\anaconda3\envs\sandbox\lib\site-packages\scipy\signal\_peak_finding.py", line 191, in argrelmax
    return argrelextrema(data, np.greater, axis, order, mode)
  File "C:\Users\shinp\anaconda3\envs\sandbox\lib\site-packages\scipy\signal\_peak_finding.py", line 245, in argrelextrema
    results = _boolrelextrema(data, comparator,
  File "C:\Users\shinp\anaconda3\envs\sandbox\lib\site-packages\scipy\signal\_peak_finding.py", line 74, in _boolrelextrema
    results &= comparator(main, plus)
numpy.core._exceptions._UFuncNoLoopError: ufunc 'greater' did not contain a loop with signature matching types (dtype('<U23'), dtype('<U23')) -> dtype('bool')

その場合は

with open('e_center.txt') as f:
    e_center_raw = [s.strip() for s in f.readlines()]

を

with open('e_center.txt') as f:
    e_center_raw = [int(s.strip()) for s in f.readlines()]

とすれば動きます。

データに
'1234 67890 234567 90123'のようなものが入っているなら、さらに変更が必要です。

Comments

[ShibaSamo]

int(s.strip())の部分をfloat(s.strip())とすることで求めていたことができました！
ありがとうございます！