scipy 将argrelextrema用于数组时的元组输出

wdebmtf2  于 2022-11-10  发布在  其他
关注(0)|答案(1)|浏览(116)

我试图获得一个数组的局部极大值的索引,a

import numpy as np
argrelextrema(a,np.less)

输出为以下元组:

(array([  maxima_1,  maxima_2, maxima_3, ...,maxima_n]),)

其中,maxima_i表示每个最大值的索引。我想简单地将其转换为一个包含以下元素的数组:maxima_1, maxima_2,...,maxima_n我尝试过使用np.asarray,但这会使我得到一个包含一个元素中所有索引的数组。

ua4mk5z4

ua4mk5z41#

检查以下代码(第3个打印值):

from scipy.signal import argrelextrema
import numpy as np 
a = np.array([2, 1, 2, 3, 2, 0, 1, 0])
print('Defulat Output{}, Numpy Array Output {}, Array output = {}'.format(argrelextrema(a,np.less), np.asarray(argrelextrema(a,np.less)), list(argrelextrema(a,np.less)[0]) ))

输出量:

相关问题