计算列表前几位元素的和

import numpy as np

 

top_k=3

arr = np.array([1, 3, 2, 4, 5])

top_k_idx=arr.argsort()[::-1][0:top_k]

print(top_k_idx)