NumPy配列で合計が最大のn個の列を抽出する

Question

100行と1000列のNumPy行列があるとします。

n列で構成された新しい行列を、元の行列の合計が最大になるように取得するにはどうすればよいですか？

Answer 1

np.argsortは@NPE hereのように使用できます。 2つの小さな配列の例を次に示します。

def nlargest_cols(a, n): 
    return a[:, sorted(a.sum(axis=0).argsort()[-n:][::-1])] 

# `a` is a 3x4 array with column sums getting 
#  larger from left to right. 
a = np.arange(12).reshape(3,4) 

# `b` is `a` rotated 2 turns. 
b = np.rot90(a, 2) 

print(nlargest_cols(a, 2)) 
# [[ 2 3] 
# [ 6 7] 
# [10 11]] 

print(nlargest_cols(b, 3)) 
# [[11 10 9] 
# [ 7 6 5] 
# [ 3 2 1]]