您可以使用
np.partition与链接的问题相同的方式：排序已经沿最后一个轴进行：
In [2]: a = np.array([[ 5, 4, 3, 2, 1], [10, 9, 8, 7, 6]])In [3]: b = np.partition(a, -3) # top 3 values from each rowIn [4]: b[:,-3:]Out[4]: array([[ 3, 4, 5], [ 8, 9, 10]])
您可以使用
np.partition与链接的问题相同的方式：排序已经沿最后一个轴进行：
In [2]: a = np.array([[ 5, 4, 3, 2, 1], [10, 9, 8, 7, 6]])In [3]: b = np.partition(a, -3) # top 3 values from each rowIn [4]: b[:,-3:]Out[4]: array([[ 3, 4, 5], [ 8, 9, 10]])
名师互学网 版权所有 (c)2021-2022 ICP备案号：晋ICP备2021003244-6号
