Сартаванне масіва Numpy паводле сума

Я хачу сартаваць масіў numpy паводле сумы. Нешта накшталт

import numpy as np
a = np.array([1,2,3,8], [3,0,2,1])
b = np.sum(a, axis = 0)
idx = b.argsort()

Цяпер np.take (a, idx) вядзе да [2, 1, 3, 8].

Але я хацеў бы масіў: result = np.array ([2, 1, 3, 8], [0, 3, 2, 1]]

Які найбольш разумны і хуткі спосаб зрабіць гэта ?