基于子集列表的排序列表（无嵌套循环）

Question

我正在尝试根据主列表对一长串标签进行排序，并且正在努力有效地实现这一点。如下面的示例所示，我想将长列表中共享字符串公共开头的所有项组合在一起，然后以与“主列表”相同的顺序创建一个新列表。你知道吗

对于Python，我通常会尝试对数组进行矢量化和并行处理，但对于基于字符串的数组来说，这似乎效果不太好。你知道吗

下面是我使用嵌套循环的示例和解决方案：

fruits = ['apple', 'banana', 'orange']  # "master list"
labels = ['banana2', 'apple2', 'orange1', 'banana1', 'apple1', 'apple3']  # "long list"
new_list = []
for fruit in fruits:
    for label in labels:
        if fruit in label:
            new_list.append(label)

print(new_list)然后返回

['apple2', 'apple1', 'apple3', 'banana2', 'banana1', 'orange1']

没有嵌套循环可以做到这一点吗？你知道吗

另外，我希望标签也能根据字符串中的最后一个数字进行排序，例如给出结果：['apple1', 'apple2', 'apple3', 'banana1', 'banana2', 'orange1']

Answer 1

这里有一种方法，使用^{}导出排序顺序。你知道吗

fruits = ['apple', 'banana', 'orange']
labels = ['banana2', 'apple2', 'orange1', 'banana1', 'apple1', 'apple3']

res = sorted(labels, key=lambda x: fruits.index(x[:-1]))

# ['apple2', 'apple1', 'apple3', 'banana2', 'banana1', 'orange1']

也可以使用tuple作为排序键，并结合一些正则表达式来分隔任意大小的整数。你知道吗

import re

fruits = ['apple', 'banana', 'orange']
labels = ['banana2', 'apple2', 'orange1', 'banana1', 'apple10', 'apple3']

def sorter(mystr, fruits):
    str_split = re.match(r'([a-z]+)([0-9]+)', mystr, re.I).groups()
    return (fruits.index(str_split[0]), int(str_split[1]))

res2 = sorted(labels, key=lambda x: sorter(x, fruits))

# ['apple2', 'apple3', 'apple10', 'banana1', 'banana2', 'orange1']