有没有更短的方法可以创建这个列表？

问题描述

这是我在尝试确定某些结果的峰值时获得的元组：

In:  peaks,_ = find_peaks(probsuc,width=4)
     peak_prominences(probsuc,peaks,wlen=7)
Out: (array([0.015625,0.24166667]),array([ 9,36]),array([14,42]))

我只想获得两个列表：[9,14] 和 [36,42]。我所做的是

In:  [peak_prominences(probsuc,wlen=7)[1][0],peak_prominences(probsuc,wlen=7)[2][0]]
Out: [9,14]

有没有更短的方法可以获得相同的结果？谢谢！！

解决方法

您确实应该避免两次调用 peak_prominences。

例如：

[arr[0] for arr in peak_prominences(probsuc,peaks,wlen=7)[1:]]

或者：

next(zip(*peak_prominences(probsuc,wlen=7)[1:]))

后者是一个元组而不是一个列表。如果您需要一个列表，则对其应用 list()。