获取列表上 max()/min() 返回的最大/最小值的索引
当使用 Python 的 max() 和对于像 minimax 这样的算法,列表上的 min() 函数通常需要确定返回的最大值或最小值的索引。
问题:
你需要找到列表中与返回的 max() 或 min() 值相对应的元素的索引,以识别哪个移动产生了该值。
解决方案:
到检索 min() 值的索引,使用以下代码:
<code class="python">index_min = min(range(len(values)), key=values.__getitem__)</code>
此方法消除了对运算符或枚举等模块的需求,并且它也优于涉及 itemgetter() 的解决方案。
对于 numpy 数组,考虑使用:
<code class="python">import numpy as np index_min = np.argmin(values)</code>
如果列表足够大并且内存消耗可以接受,这会更快。
基准:
下面的基准测试比较了两种建议解决方案的效率:
[图像显示了不同方法的运行时间图表]
蓝线代表纯 Python 解决方案,而红线使用numpy。黑线是使用 itemgetter() 的参考实现。
对于大型列表,numpy 解决方案明显更快。然而,对于较小的列表,纯 Python 解决方案可能更有效。
以上是如何高效地查找Python列表中最大/最小值的索引?的详细内容。更多信息请关注PHP中文网其他相关文章!