在使用python处理数据时可能会遇到根据列表A对列表B进行排序的问题,记录一下想到的两个方法。
方法1:根据列表b中每个元素的下标来获取列表a中对应位置的元素,将其作为排序依据即可。
import random a = [x for x in range(5)] b = [x for x in range(5,10)] random.shuffle(a) random.shuffle(b) print(a) print(b) # sort list b by list a sorted_b = sorted(b, key=lambda x: a[b.index(x)]) print(sorted_b)
方法2:输出为:
[1,0,4,3,2]
[5,7,9,8,6]
[7,5,6,8,9]
用zip函数将两个列表打包,根据a排序,再解包即可。
import random a = [x for x in range(5)] b = [x for x in range(5,10)] random.shuffle(a) random.shuffle(b) print(a) print(b) zip_a_b = zip(a,b) sorted_zip = sorted(zip_a_b, key=lambda x:x[0]) sorted_a, sorted_b = zip(*sorted_zip) print(list(sorted_a)) print(list(sorted_b))
输出为:
[4, 1, 0, 2, 3]
[6, 5, 9, 8, 7]
[0, 1, 2, 3, 4]
[9, 5, 8, 7, 6]



