对于元组列表中的元素,返回元组中的其他元素

Question

下面是我的函数代码,用于搜索名为pair的元组列表.列表中的每个元组由两个元素组成.

函数中给出的元素(称为item)可以位于元组的0或1位置,但列表中的所有元组仅由2个元素组成.

pairs = [('a','b'),('c','d'),('e','f'),('c','a')]

def connections(pairs, item):
    output = ''
    for j in pairs:
        if item in j:
            output = output + str(j)
    return (output)

现在,这段代码将拉出包含其中项目的整个元组.

我想更改此代码只返回元组中的其他元素,并将其作为字符串而不是元组返回.

Answer 1

这应该做你想要的:

def connection(pairs, item):
    return ''.join([p[1-p.index(item)] for p in pairs if item in p])

这是一些示例输出:

>>> connection([('a','b'),('c','d'),('e','f'),('c','a')], 'a')
'bc'