https://thaothuc.com/nested-if-else-in-list-comprehension-python