我有一个多维 numpy 数组，由如下元组组成：
[[(0.56, 1),(0.25, 4), ...],[(0.11, 9), ...], ...]
每个元组的第二个元素是索引引用。我想提取每行具有最高第一个值的元组。有没有办法用 numpy max 实现这一目标？
我尝试过的一件事是使用轴参数，如下所示：
np.max(my_array, axis=0)
但这会在未保留索引引用的情况下围绕对进行洗牌。例如，上面示例中的第一行会显示类似的内容，[(0.56,4), ...]而我希望它显示[(0.56,1), ...]
在普通的python中，您可以使用：
[max(row, key=lambda row: row[0]) for row in array]
本文收集自互联网，转载请注明来源。
如有侵权，请联系 [email protected] 删除。
我来说两句