Для элемента в списке кортежей, возвращаем другой элемент в кортеж

Question

Ниже приведен мой код для функции, которая ищет список кортежей, называемых парами. Каждый кортеж в списке состоит из двух элементов.

Элемент, заданный в функции (называемый элементом), может находиться в 0 или 1 позиции кортежа, но все кортежи в списке состоят всего из двух элементов.

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' 

Answer 2

Вместо проверки членства проверьте равенство.

def connections(pairs, item): 
    output = '' 
    for j in pairs: 
     if item == j[0]: 
      output += j[1] 
     elif item == j[1]: 
      output += j[0] 
    return output 

Обратите внимание, что я упростил конкатенацию в дополненное задание и удаляюсь ненужные str() вызовов, так как они уже строка. Или вы могли бы сделать это следующим образом:

def connections(pairs, item): 
    return ''.join(j[0] if item==j[1] else j[1] if item==j[0] else '' for j in pairs) 

Answer 3

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

def connections(pairs, item): 
    output = '' 
    for j in pairs: 
     if item in j: 
      index = 1-j.index(item) 
      output = output + str(j[index]) 
    return (output) 

print(connections(pairs, 'c')) 

Выход:

da