Python 3.8中高效遍历多列表并记录各索引最大值所属列表的方法

针对你15个同长度浮点列表的需求，完全可以避免繁琐的if判断，以下两种简洁高效的实现方式：

方法一：纯Python原生实现（无需额外依赖） #

先把所有列表统一放到一个容器里，利用zip将每个索引位置的元素打包成元组，再通过列表推导式遍历每个元组，找到最大值对应的原列表序号（序号从1开始）：

# 将所有列表存入一个列表中，方便统一处理
all_lists = [List1, List2, List3]  # 实际使用时替换为你的15个列表
# 遍历每个索引位置的元素组，找到最大值对应的列表序号
result = [group.index(max(group)) + 1 for group in zip(*all_lists)]

运行示例中的三个列表，会得到result = [1, 1, 1, 3, 3]，符合预期。

方法二：用Numpy实现（适合大数据量场景） #

如果你的列表长度很长，Numpy的向量化操作会比纯Python更快：

import numpy as np

# 将列表转换为Numpy数组，每一行对应一个原始列表
arr = np.array([List1, List2, List3])
# 沿垂直方向取每个位置最大值的索引，再加1转为从1开始的序号
result = (np.argmax(arr, axis=0) + 1).tolist()

注意事项 #

• 如果多个列表在同一索引位置的值相同且都是最大值，上述两种方法都会返回第一个出现该最大值的列表序号，符合常规业务逻辑。
• 纯Python方法无需安装额外库，适合轻量场景；Numpy方法在处理超大长度列表或大量列表时，性能优势明显。

内容的提问来源于stack exchange，提问作者Tritize