本文旨在介绍如何从一个Python列表中交替提取最大值和最小值，并获取它们对应的索引。我们将探讨两种基于`itertools`库的解决方案，通过分组和累积计算，高效地实现这一目标，并提供详细的代码示例和解释，帮助读者理解和应用这些技巧。
在处理数据时，有时需要从列表中提取特定模式的值。一个常见的需求是交替提取最大值和最小值，例如，先找到列表中的最大值，然后找到该最大值之后的最小值，接着找到该最小值之后的最大值，以此类推。本文将介绍两种使用Python的itertools库来实现这一目标的有效方法。
方法一：基于groupby直接提取值和索引
此方法的核心思想是使用itertools.groupby将列表分成递增和递减的子序列。然后，从每个子序列的最后一个元素中提取值和索引。
from itertools import groupby
myList = [10, 12, 18, 20, 25, 18, 17, 16, 10, 20, 30, 35, 40, 35, 30, 20, 15]
d = [(myList[i], i) for _, (*_, i) in groupby(range(1, len(myList)),
key=lambda i: myList[i - 1] < myList[i])]
print(d) # 输出: [(25, 4), (10, 8), (40, 12), (15, 16)]代码解释：
立即学习“Python免费学习笔记（深入）”；
macOS UI 自动化技能包 —— 用 Python 控制你的 Mac
下载
让 macOS 自动化像说话一样简单 —— 用代码控制你的 Mac，无需学习 AppleScript 系统要求 macOS 14+ (Sonoma) Python 3.9+ 需要开启：辅助功能 + 屏幕录制权限
- groupby(range(1, len(myList)), key=lambda i: myList[i - 1]
- (*_, i)：这部分代码解包每个分组。_用于忽略分组中除了最后一个元素之外的所有元素。i是最后一个元素的索引。
- (myList[i], i)：这部分代码创建一个元组，包含列表中索引i处的值和索引i本身。
- 整个列表推导式构建了一个包含交替最大值和最小值及其索引的列表。
方法二：先计算索引，再提取值
此方法首先计算出交替最大值和最小值的索引，然后再从列表中提取这些索引对应的值。
from itertools import groupby, accumulate myList = [10, 12, 18, 20, 25, 18, 17, 16, 10, 20, 30, 35, 40, 35, 30, 20, 15] d = (b > a for a, b in zip(myList, myList[1:])) *indexes, = accumulate(len(g) for _, (*g,) in groupby(d)) values = [myList[i] for i in indexes] print(indexes) # 输出: [4, 8, 12, 16] print(values) # 输出: [25, 10, 40, 15]
代码解释：
立即学习“Python免费学习笔记（深入）”；
- d = (b > a for a, b in zip(myList, myList[1:]))：这部分代码创建一个生成器，用于判断列表中相邻元素的大小关系。如果后一个元素大于前一个元素，则返回True，否则返回False。
- *indexes, = accumulate(len(g) for _, (*g,) in groupby(d))：这部分代码使用groupby函数对大小关系生成器进行分组，并使用accumulate函数计算每个分组的长度的累积和。accumulate的结果就是交替最大值和最小值的索引。
- values = [myList[i] for i in indexes]：这部分代码使用列表推导式从列表中提取索引indexes对应的值。
注意事项
- 这两种方法都依赖于itertools库，该库提供了许多高效的迭代器工具。
- 这两种方法都假设列表至少包含两个元素。如果列表为空或只包含一个元素，则需要进行额外的处理。
- 第一种方法在一次迭代中同时提取值和索引，可能更简洁。
- 第二种方法将索引计算和值提取分开，可能更易于理解和调试。
总结
本文介绍了两种使用itertools库从Python列表中交替提取最大值和最小值及其索引的方法。第一种方法使用groupby直接提取值和索引，而第二种方法先计算索引，再提取值。选择哪种方法取决于具体的需求和偏好。通过学习这些方法，可以更加高效地处理列表数据，并提取所需的模式。
