Notes on Python Comprehensions

List Comprehensions

x = [i for i in range(10)]

if [i for i in line if "SOME TERM" in i]:
    # do something

x = ['1', '2', '3', '4', '5']
y = [int(i) for i in x]

my_strs = [s.strip() for s in str_list]

vec = [[1, 2, ,3], [4, 5, 6], [7, 8, 9]]
[num for elem in vec for num in elem] # [1, 2, 3, ..., 9]

Dicitonary Comprehensions

{i : str(i) for i in range(10)}

my_dict = {1 : "dog", 2 : "cat", 3 : "hamster"}
d = {value : key for key, value in my_dict.items()}

Set Comprehensions

my_list = [1, 2, 2, 3, 4, 5, 5, 7, 8]
my_set = {x for x in my_list}