首页  >  文章  >  后端开发  >  如何高效地查找Python列表中最大/最小值的索引?

如何高效地查找Python列表中最大/最小值的索引?

Barbara Streisand
Barbara Streisand原创
2024-10-29 18:24:14628浏览

 How to Efficiently Find the Index of Max/Min Values in Python Lists?

获取列表上 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中文网其他相关文章!

声明:
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn