从 Python 中的数字列表中查找局部峰值元素索引的程序

假设我们有一个名为nums的数字列表，其长度至少为2。我们必须找到列表中每个峰值的索引。该列表按升序排序。指数i是峰值，当-

nums[i]>nums[i+1]当i=0

nums[i]>nums[i-1]当i=n-1

nums[i-1]<nums[i]>nums[i+1]else

因此，如果输入类似于nums=[5,6,7,6,9]，那么输出将是[2,4]，因为索引2处的元素是7，它大于两个邻居，并且项目位于索引4是9，这比它的左项大。

示例

让我们看下面的实现来更好地理解

def solve(nums): ans = [] n = len(nums) if n == 1: return ans for i, num in enumerate(nums): if i > 0 and i < n - 1: if nums[i - 1] < num > nums[i + 1]: ans.append(i) if i == 0: if num > nums[i + 1]: ans.append(i) if i == n - 1: if num > nums[i - 1]: ans.append(i) return ans nums = [5, 6, 7, 6, 9] print(solve(nums))

输入

[5, 6, 7, 6, 9]输出结果[2, 4]