Python 根据精确的组件对numpy.array的行进行排序
我有一个如下所示的数组Python 根据精确的组件对numpy.array的行进行排序,python,numpy,Python,Numpy,我有一个如下所示的数组 import numpy as np a=np.array([[3.1,10,6.],[7.,4,8.1]]) 我想对行进行重新排序,使行的第二个元素增加。所以,我想要以下输出 [[ 7. 4. 8.1] [ 3.1 10. 6. ]] 因为4np.argsort将返回对1D数组进行排序的索引。 因此,您希望使用将对a[:,1]进行排序的索引对a进行排序,即a中每行的第二个元素 你可以用它来做这个 a[np.argsort(a[:,1])] # arr
import numpy as np
a=np.array([[3.1,10,6.],[7.,4,8.1]])
我想对行进行重新排序,使行的第二个元素增加。所以,我想要以下输出
[[ 7. 4. 8.1]
[ 3.1 10. 6. ]]
因为
4np.argsort
将返回对1D数组进行排序的索引。
因此,您希望使用将对a[:,1]
进行排序的索引对a
进行排序,即a
中每行的第二个元素
你可以用它来做这个
a[np.argsort(a[:,1])]
# array([[ 7. , 4. , 8.1],
# [ 3.1, 10. , 6. ]])
a[np.argsort(a[:,1])]
# array([[ 7. , 4. , 8.1],
# [ 3.1, 10. , 6. ]])