我有一个长长的Python元组,我想在其中找到最大的价值。我有一些数值和它们的概率,元组样本是这样的:
(array([ 5, 5, 3, ..., -1, -4, --5]), array([ 0.25, 0.75, 0.8, ...0.05, 0.2, 0.6])
我想要的是能够提取最大概率,但获得相应的数值(X),从而基本上创建一个新的元组。根据上面的数字,类似这样的东西:
> Max tuple: ([3,0.8])
我已经做了这样的事情例如new_tuple = max(map(max, tuple_list))[1]
,但它只给我最高的概率,而不是它对应的数字,我如何提取最高的第二个数(概率)来创建一个新的元组?
您可以压缩两个元素并使用自定义键获取最大值。
您可以将两个数组一起zip
,然后将key
函数传递给max
以选择第二个数组中具有最大元素的对。
您可以做: