Простые передовые практики в Python [closed]

Я пишу код для некоторых оценок кодирования и хотел бы получить совет о том, что можно считать более профессиональным кодом. Я надеюсь, что это нормальный вопрос — я впервые в Codereview SE.

Следующие две (фиктивные) функции делают то же самое, но я ввел несколько вариантов:

• Способ проверки условия (что, если бы я хотел проверить 3+ равенства?)
• В else утверждение

Что будет считаться наилучшим подходом к этим деталям?

def function(n):
    if n == 0 or n == 1:
        return 1
    else:
        print("Not 0 or 1")
        return 5

def function(n):
    if n in [0,1]:
        return 1
        
    print("Not 0 or 1")
    return 5