从 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]