Для начало поймем что должна делать функция
если вводиться 1 возвращать 2
если вводиться 2 возвращать 1 (или иначе возвращать 1)
напишем код
def get_b(a):
if a == 1:
return 2
else: # else не обязателен, т.к. return прекращает выполнение
# функции, но для большей понятности проги он нам нужен
return 1
#теперь простая проверка функции (ответ у нас уже есть)
print ("если а = 1, б =", get_b(1))
print ("если а =2, б =", get_b(2))
P.S. вопросы в коменты