Get maximum subset in multidimensional array [closed]

September 19, 2022 September 19, 2022 / 1 minute of reading

You can sum along columns and then find the index with the maximum value with argmax:

a[np.argmax(a.sum(axis=1))]

CLICK HERE to find out more related problems solutions.