python - 对列表进行排序，其中某些值保持在固定位置

Question

我有一个字符串列表。我只想对满足特定条件的值进行排序。考虑这个列表

['foo','bar','testa','python','java','abc']

我只想对其中包含 a 的值进行排序。结果应该是这样的

['foo','abc','bar','python','java','testa']

带有a的元素会适当改变位置，但其他元素保持原来的位置。

我完全不知道如何实现它，所以我希望其他人能做到。谁能告诉我怎么做？

Answer 1

y = sorted(w for w in x if 'a' in w)  # pick and sort only the elements with 'a'
x = [w if 'a' not in w else y.pop(0) for w in x]

最后一行保留没有 'a' 的单词不变，而那些带有 'a' 的单词从 y 列表中逐步挑选(已经排序)

编辑: @MartijnPieters 解决方案性能更好，因为它使用迭代器并且不会使用额外的内存来存储 y。

y = iter(sorted(w for w in x if 'a' in w))  # create iterator, don't use memory
x = [w if 'a' not in w else next(y) for w in x]  # yield from iter instead of popping from a list

因为看起来你需要这个算法来处理不同的条件，你可以把它放到一个方法中:

x = ['foo','bar','testa','python','java','abc']

def conditional_sort(ls, f):
    y = iter(sorted(w for w in ls if f(w)))
    return [w if not f(w) else next(y) for w in ls]

conditional_sort(x, lambda w: 'a' in w)

第一个参数是列表，第二个参数是一个函数，它接受一个参数并返回一个 bool 值。