问题描述:假如我们有两组数据,是一一对应的关系,例如:3.5->7, 2.6->6 ...
现在我们需要对一组数据进行排序。排序后找到另一组数据与其对应的数值。假设两组数据的对应关系如下,我们需要对list2进行排序,排序后查看原来其对应的list1中数据是什么。
list1 = [3.5, 2.6, 4.3, 5.2, 13.14, 15.20]
list2 = [7, 6, 8, 50, 335, 2]
1.方法一:迭代器
from operator import itemgetter
list1 = [3.5, 2.6, 4.3, 5.2, 13.14, 15.20]
list2 = [7, 6, 8, 50, 335, 2]
# 方法一:迭代器
en_list = list(enumerate(list2)) # 放入迭代器后的列表
so_list = sorted(en_list, key=itemgetter(1)) # 对值进行排序后的列表
# 如果我们想取list2中最小值的索引对应list1的值
list1[b[0][0]]
结果为15.2,正确!
2.方法二:numpy
import numpy as np
list1 = [3.5, 2.6, 4.3, 5.2, 13.14, 15.20]
list2 = [7, 6, 8, 50, 335, 2]
# 还是取最小的值索引
so_list = np.argsort(list2)
print(list1[so_list[0]])
结果同样为15.2,正确!
除此以外,还可以将两组数据放入字典,通过key-value的方式进行查询,或者使用pandas的DataFrame。
如果大家有更好的方法,欢迎留言讨论!