鉴于：
a=np.array([[-0.00365169, -1.96455717, 1.44163783, 0.52460176, 2.21493637],
[-1.05303533, -0.7106505, 0.47988974, 0.73436447, -0.87708389],
[-0.76841759, 0.8405524, 0.91184575, -0.70652033, 0.37646991]])
我想获得最大子集（在这种情况下，第一行）：
[-0.00365169, -1.96455717, 1.44163783, 0.52460176, 2.21493637]
通过使用print(np.amax(a, axis=0))，我得到了错误的结果：
[-0.00365169 0.8405524 1.44163783 0.73436447 2.21493637]
我们如何获得正确的最大子集？
您可以sum沿列查找，然后使用以下命令找到具有最大值的索引argmax：
a[np.argmax(a.sum(axis=1))]
本文收集自互联网，转载请注明来源。
如有侵权，请联系 [email protected] 删除。
我来说两句