我想按第一行对numpy中的数组进行排序。
例如 :
import numpy as np
test = np.array([[1334.71601720318, 930.9757468052002, 1018.7038817663818],
[0.0, 1.0, 2.0],
[ np.array([[ 667, 1393],
[1961, 474]]),
np.array([[ 673, 1389],
[ 847, 1280]]),
np.array([[ 726, 1077],
[ 898, 961]])]], dtype=object)
我想对行进行排序:
[1334.71601720318, 930.9757468052002, 1018.7038817663818]
获得:
np.array([[930.9757468052002, 1018.7038817663818, 1334.71601720318],
[1.0, 2.0 ,0.0],
[ np.array([[ 673, 1389],
[ 847, 1280]]),
np.array([[ 726, 1077],
[ 898, 961]])],
np.array([[ 667, 1393],
[1961, 474]])], dtype=object)
----编辑----
我尝试了:sorted(test,key = lambda row:row [1])但我收到一条错误消息:“具有多个元素的数组的真值不明确。请使用a.any()或a.all ()”
我认为test[:, np.argsort( test[0] ) ]
应该可以解决问题。
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句