Python List Comprehensions

Python List Comprehensions

basic format: new_list = [transform sequence [filter] ]
© Joe James, 2019.

import random

get values within a range

under_10 = [x for x in range(10)]
print('under_10: ' + str(under_10))

under_10: [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

get squared values

squares = [x**2 for x in under_10]
print('squares: ' + str(squares))

squares: [0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

get odd numbers using mod

odds =  [x for x in range(10) if x%2 == 1]
print('odds: ' + str(odds))

odds: [1, 3, 5, 7, 9]

get multiples of 10

ten_x = [x * 10 for x in range(10)]
print('ten_x: ' + str(ten_x))

ten_x: [0, 10, 20, 30, 40, 50, 60, 70, 80, 90]

get all numbers from a string

s = 'I love 2 go t0 the store 7 times a w3ek.'
nums = [x for x in s if x.isnumeric()]
print('nums: ' + ''.join(nums))

nums: 2073

get index of a list item

names = ['Cosmo', 'Pedro', 'Anu', 'Ray']
idx = [k for k, v in enumerate(names) if v == 'Anu']
print('index = ' + str(idx[0]))

index = 2

delete an item from a list

letters = [x for x in 'ABCDEF']
random.shuffle(letters)
letrs = [a for a in letters if a != 'C']
print(letters, letrs)

['D', 'F', 'E', 'A', 'B', 'C'] ['D', 'F', 'E', 'A', 'B']

if-else condition in a comprehension
must come before iteration.

nums = [5, 3, 10, 18, 6, 7]
new_list = [x if x%2 == 0 else 10*x for x in nums]
print('new list: ' + str(new_list))

new list: [50, 30, 10, 18, 6, 70]

nested loop iteration for 2D list
b is the subsets, x is the values.

a = [[1,2],[3,4]]
new_list = [x for b in a for x in b]
print(new_list)

[1, 2, 3, 4]