python - 比较列表列表中的元素

Question

我基本上有一个这样的列表列表： [['sd', 'pd', 'od'], ['sd', 'pd1', 'od2'], ['sd2', 'pd2', 'od']] 我希望输出是['sd pd od ; pd1 od2 .', 'sd2 pd2 od .'] 所以，我想比较每个列表的第一个元素，如果它们相同，我想省略第一个元素（发生在第二个列表中）。

这是我的代码，它不起作用，因为它比较元素太多，我不知道为什么。

def simplistic(triples):
base = []
for x in triples:
    for y in triples:
        if x[0] == y[0]:
            base.append((x[:],y[1],y[2]))
print(base)

这段代码的输出是：

[(['sd', 'pd', 'od'], 'pd', 'od'), (['sd', 'pd', 'od'], 'pd1', 'od2'), (['sd', 'pd1', 'od2'], 'pd', 'od'), (['sd', 'pd1', 'od2'], 'pd1', 'od2'), (['sd2', 'pd2', 'od'], 'pd2', 'od')]

Answer 1

由于您的问题有点不清楚，请尝试此操作并检查是否可以。

bad_list = []
f=''
for i,s in enumerate(l):
    if s[0] == l[i-1][0] or s[0] in bad_list:
        f = f + ' '.join(s[1:])
        bad_list.append(s[0])           
    else:
        f = f + ' '.join(s)
    f = f+ ' ;'
f = [f[:-1] + '.']

这f将是您的预期结果。

Answer 2

如果您正在寻找基于第一个元素对列表进行分组，您可以使用defaultdict

>>> from collections import defaultdict
>>> triples =  [['sd', 'pd', 'od'], ['sd', 'pd1', 'od2'], ['sd2', 'pd2', 'od']]
>>> d = defaultdict(list)
>>> 
>>> for k,*l in triples:
...     d[k].extend(l)
...
>>> res = [[k]+l for k,l in d.items()]
>>> print(res)
[['sd', 'pd', 'od', 'pd1', 'od2'], ['sd2', 'pd2', 'od']]