Для начало поймем что должна делать функция

если вводиться 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. вопросы в коменты