Python 3 cheat sheet

• pipenv
• pandas

List Comprehension

list = [1, 2, 3, 4, 5]

a = []
for i in list:
    if i > 3:
        a.append(i**2)
        
>>> a
[16, 25]
        
a = [i**2 for i in list if i > 3]

>>> a
[16, 25]

Dict Comprehension

dict = {'a': 1, 'b': 2, 'c': 3, 'd': 4}

d = {}
for k,v in dict.items():
    if v > 2:
        d[k] = v

>>> d
{'c': 3, 'd': 4}

d = {k: v for k,v in dict.items() if v > 2}

>>> d
{'c': 3, 'd': 4}

Nested Dict Comprehension

shift = {}
for row in ws.iter_rows(f"B{ws.min_row}:B{ws.max_row}"):
    for cell in row:
        if cell.value is not None:
            shift[cell.value] = cell.row
            
shift = {cell.value: cell.row for row in ws.iter_rows(f"B{ws.min_row}:B{ws.max_row}") for cell in row if cell.value is not None}

Classes:

class Person():
    def __init__(self, name, birthDate):
        self.name = name.lower().capitalize()
        self.birthDate = birthDate
        
    def jump(self):    
        return f"{self.name} is jumping!"

>>> patrick = Person('patrick', '1-1-1980')
>>> patrick.name
'Patrick'
>>> patrick.jump()
'Patrick is jumping'
>>> patrick.birthDate
'1-1-1980'