Comprehensions
Python para usuários de R
Daniel Chen
Instructor
Comprehensions são loops
- Iterar (loop) por uma lista
- Executar alguma função
- Adicionar os resultados a uma nova lista
List comprehension
Loop
data = [1, 2, 3, 4, 5]
new = []
for x in data:
new.append(x**2)
print(new)
[1, 4, 9, 16, 25]
Comprehension
data = [1, 2, 3, 4, 5]
new = [x**2 for x in data]
print(new)
[1, 4, 9, 16, 25]
Dictionary comprehension
Loop
data = [1, 2, 3, 4, 5]
new = {}
for x in data:
new[x] = x**2
print(new)
{1: 1, 2: 4, 3: 9, 4: 16, 5: 25}
Comprehension
data = [1, 2, 3, 4, 5]
new = {x: x**2 for x in data}
print(new)
{1: 1, 2: 4, 3: 9, 4: 16, 5: 25}
Alternativas ao loop `for`
- Loop
forpara iterar - R: funções
sapply,lapply,apply - Python: função
mape métodoapply
Map
Loop For
def sq(x):
return x**2
l = [1, 2, 3]
for i in l:
print(sq(i))
1
4
9
Map
map(sq, l)
<map at 0x7f41cca38358>
list(map(sq, l))
[1, 4, 9]
Vamos praticar!
Python para usuários de R
