我有两个包含numpy数组的列表Ii0和Iv0.我用argsort()对Ii0中的元素重新排序以生成新的列表Ii01.我想用相同的顺序对Iv0中的元素进行重新排序以产生新的列表Iv01.我给出了当前和预期的输出.
import numpy as np
Ii0 = [np.array([[0, 1],[0, 2],[1, 3],[2, 4],[4,3]]),
np.array([[0, 1],[0, 2],[1, 3],[2, 5],[4,3],[3,4]])]
Iv0 = [np.array([[10],[20],[30],[40],[50]]),
np.array([[25],[38],[41],[97],[65],[54]])]
Ii01 = [i[i[:,1].argsort()] for i in Ii0]
Iv01 = [i[i[:,0].argsort()] for i in Iv0]
print("Iv01 =",Iv01)
电流输出为
Iv01 = [array([[10],
[20],
[30],
[40],
[50]]), array([[25],
[38],
[41],
[54],
[65],
[97]])]
预期输出为
Iv01 = [array([[10],
[20],
[30],
[50],
[40]]),array([[25],
[38],
[41],
[65],
[54],
[97]])]