有没有办法使用键(或比较器)函数对numpy ndarray的行进行排序,而无需转换为python列表？
特别是,我需要根据这个功能排序：
c1,c2= 4,7
lambda row: c1*(row[1]/c2)+row[0]
我意识到一个可能的解决方案是生成一个具有每行键值的向量,但是如何根据它进行排序？是否应该以某种方式将这种向量转换为索引向量？
order= c1*(matrix[:,1]/c2)+matrix[:,0]
indexes= order_to_index( order )
return matrix[ indexes ]
这是现实的吗？
最佳答案
你的方法是对的,它类似于Schwartzian transform或Decorate-Sort-Undecorate (DSU) idiom
正如我所说,你可以使用numpy函数np.argsort.它完成你的order_to_index的工作.