如何在python中对列表中的两个列表路径的字符串进行排序？

Question

我有两个包含文件路径的列表

lst_A =['/home/data_A/test_AA_123.jpg',
        '/home/data_A/test_AB_234.jpg',
        '/home/data_A/test_BB_321.jpg',
        '/home/data_A/test_BC_112.jpg',
       ]

lst_B =['/home/data_B/test_AA_222.jpg',
        '/home/data_B/test_CC_444.jpg',
        '/home/data_B/test_AB_555.jpg',
        '/home/data_B/test_BC_777.jpg',
       ]

基于lst_A ，我想对列表B进行排序，以使A和B中两个路径的基名的名字和名字应该相同。 在这种情况下是test_xx 。 因此，预期的候选清单B为

lst_B =['/home/data_B/test_AA_222.jpg',
        '/home/data_B/test_AB_555.jpg',
        '/home/data_B/test_CC_444.jpg',
        '/home/data_B/test_BC_777.jpg',
       ]

另外，我想指出两个列表中哪个位置的名字中的名字和名字相同（例如test_xx ），因此数组指示符应为

array_same =[1,1,0,1]

我应该如何在python中做到这一点？ 我已经尝试过.sort（）函数，但是它返回了意外的结果。 谢谢

更新：这是我的解决方案

import os
lst_A =['/home/data_A/test_AA_123.jpg',
        '/home/data_A/test_AB_234.jpg',
        '/home/data_A/test_BB_321.jpg',
        '/home/data_A/test_BC_112.jpg',
       ]

lst_B =['/home/data_B/test_AA_222.jpg',
        '/home/data_B/test_CC_444.jpg',
        '/home/data_B/test_AB_555.jpg',
        '/home/data_B/test_BC_777.jpg']

lst_B_sort=[]
same_array=[]
for ind_a, a_name in enumerate(lst_A):
  for ind_b, b_name in enumerate(lst_B):
    print (os.path.basename(b_name).split('_')[1])
    if os.path.basename(b_name).split('_')[1] in os.path.basename(a_name):
        lst_B_sort.append(b_name)
        same_array.append(1)
print(lst_B_sort)
print(same_array)

Output: ['/home/data_B/test_AA_222.jpg', '/home/data_B/test_AB_555.jpg', '/home/data_B/test_BC_777.jpg']

[1, 1, 1]

因为我没有添加名称不同的元素

Answer 1

遍历lst_A ，获取文件名前缀，然后将lst_B具有相同前缀的元素追加到结果列表中。

从lst_B创建所有元素的lst_B ，然后在结果中添加路径时，将其从集合中删除。 然后，最后您可以遍历此集合，在没有匹配项的结果中填充空白。

lst_A =['/home/data_A/test_AA_123.jpg',
        '/home/data_A/test_AB_234.jpg',
        '/home/data_A/test_BB_321.jpg',
        '/home/data_A/test_BC_112.jpg',
       ]

lst_B =['/home/data_B/test_AA_222.jpg',
        '/home/data_B/test_CC_444.jpg',
        '/home/data_B/test_AB_555.jpg',
        '/home/data_B/test_BC_777.jpg',
       ]

new_lst_B = []
same_array = []
set_B = set(lst_B)
for fn in lst_A:
    prefix = "_".join(os.path.basename(fn).split('_')[:-1])+'_' # This gets test_AA_
    try:
        found_B = next(x for x in lst_B if os.path.basename(x).startswith(prefix))
        new_lst_b.append(found_B)
        same_array.append(1)
        set_B.remove(found_B)
    except StopIteration: # No match found
        new_lst_b.append(None) # Placeholder to fill in
        same_array.append(0)
for missed in set_B:
    index = new_lst_B.index(None)
    new_lst_B[index] = missed

DEMO

Answer 2

我们将使用SIMPLE技术和应用解决方案讨论该问题。

简单

我们只专注于对给定键的名称进行排序。

特定

简单名称和键列表：

lst_a = "AA AB BB BC EE".split()
lst_b = "AA DD CC AB BC".split()

key_list = [1, 1, 0, 1, 0]

码

same = sorted(set(lst_a) & set(lst_b))
diff = sorted(set(lst_b) - set(same))

isame, idiff = iter(same), iter(diff)
[next(isame) if x else next(idiff) for x in key_list]
# ['AA', 'AB', 'CC', 'BC', 'DD']

lst_b根据与lst_a共享的元素进行排序。 根据需要插入残余物。

---

细节

这个问题主要减少到对两个列表中的名称交集进行排序。 相交是一组称为same的公共元素。 残留物位于diff 。 我们的排序same和diff ，在这里就是他们的样子：

same
# ['AA', 'AB', 'BC']
diff
# ['CC', 'DD']

现在，我们只想根据键从任一列表中按顺序提取值。 我们从迭代key_list开始。 如果为1 ，则从isame迭代器中提取。 否则，请从idiff 。

现在我们有了基本技术，我们可以将其应用于更复杂的路径示例。

---

应用

将此想法应用于更复杂的路径字符串：

特定

import pathlib 


lst_a = "foo/t_AA_a.jpg foo/t_AB_a.jpg foo/t_BB_a.jpg foo/t_BC_a.jpg foo/t_EE_a.jpg".split()
lst_b = "foo/t_AA_b.jpg foo/t_DD_b.jpg foo/t_CC_b.jpg foo/t_AB_b.jpg foo/t_BC_b.jpg".split()

key_list = [1, 1, 0, 1, 0]

# Helper
def get_name(s_path):
    """Return the shared 'name' from a string path.

    Examples
    --------
    >>> get_name("foo/test_xx_a.jpg")
    'test_xx'

    """
    return pathlib.Path(s_path).stem.rsplit("_", maxsplit=1)[0]

码

将名称映射到路径：

name_path_a = {get_name(p): p for p in lst_a}
name_path_b = {get_name(p): p for p in lst_b}

名称在字典键中，因此直接用字典键替换集：

same = sorted(name_path_a.keys() & name_path_b.keys())
diff = sorted(name_path_b.keys() - set(same))

isame, idiff = iter(same), iter(diff)

通过从迭代器提取的名称获取路径：

[name_path_b[next(isame)] if x else name_path_b[next(idiff)] for x in key_list]

产量

['foo/t_AA_b.jpg',
 'foo/t_AB_b.jpg',
 'foo/t_CC_b.jpg',
 'foo/t_BC_b.jpg',
 'foo/t_DD_b.jpg']