https://arlarse.com/programming/python/conditional-selection-python/