python:根据一个列表对另外一个列表排序

在使用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)

输出为:
[1,0,4,3,2]
[5,7,9,8,6]
[7,5,6,8,9]

方法2:

用zip函数将两个列表打包,根据a排序,再解包即可。与方法1的区别在于方法2对a、b都进行了排序。

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]

猜你喜欢

转载自blog.csdn.net/zxdd2018/article/details/124714603