`import heapq
import numpy as np
import random
a = np.random.randint(50,size= (4,5))
a = np.array(a)
print(a)
lists = [[] for i in range(4)]
for i in range(len(a)):
# print(heapq.nlargest(3, range(len(a[i])), a[i].take))
lists[i].append(heapq.nlargest(3, range(len(a[i])), a[i].take))
print(lists)
`
输出结果如下：
[[14 5 29 32 2]
[34 24 49 31 35]
[39 37 43 40 5]
[30 0 42 28 8]]
[[[3, 2, 0]], [[2, 4, 0]], [[2, 3, 0]], [[2, 0, 3]]]