https://tutorial.eyehunts.com/python/python-list-comprehension-if-else-example-code/