我有一个numpy.ndarray公司具体如下:
from numpy import array
a = array( [[1,1,0.4], [1,1,0.3],[0.4,0.3,1]] )
array([[ 1. , 1. , 0.4],
[ 1. , 1. , 0.3],
[ 0.4, 0.3, 1. ]])
以下是列:
dataidx = array( [1,2,3] )
我想按行对上面数组的值进行排序,然后指定相关的dataidx:
indices = np.argsort(-a, axis=1)
result = np.hstack((dataidx[:, None], dataidx[indices]))
print(result)
[[1 1 2 3]
[2 1 2 3]
[3 3 1 2]]
对于每一行,如何根据下面的第一列排除dataidx?你知道吗
[[1 2 3]
[2 1 3]
[3 1 2]]
有一个办法-
还有一个-
相关问题 更多 >
编程相关推荐