python - 仅当部分单词尚未在列表中时才追加

Question

我有一个如下所示的列表，并想从中创建两个列表。如果第一个列表包含 S01_a，则 S01_b 应位于第二个列表中。

my_list = ['S01_a', 'S01_b', 'S02_a', 'S02_b', 'S03_a', 'S03_b', 'S04_a', 'S04_b']

我尝试了这个，但我不知道如何引用list1中项目的[1:2]字符。有人有解决办法吗？

stims = ["S01_a", "S01_b", "S02_a", "S02_b", "S03_a", "S03_b", "S04_a", "S04_b"]

l1 = []
l2 = []
for item in stims:
    if item[1:2] not in l1:
        l1.append(item)
    else:
        l2.append(item)


print(l1)
print(l2)

输出为:

['S01_a', 'S01_b', 'S02_a', 'S02_b', 'S03_a', 'S03_b', 'S04_a', 'S04_b']
[]

提前谢谢您。

Answer 1

您可以将 any() 与生成器一起使用，该生成器针对列表中的每个元素测试 item[1:3]。

for item in stims:
    if not any(item[1:3] == el[1:3] for el in l1):
        l1.append(item)
    else:
        l2.append(item)

您需要使用切片[1:3]。请记住，第二个索引不包含在内，因此 [1:2] 仅选择数字 0，而不是两位数字 01、02等