De un bucle a una sola expresión
Un bucle muy habitual empieza con una lista vacía y añade un resultado por elemento. Una comprensión de lista dice lo mismo en una sola expresión, entre corchetes: [lo_que_se_construye for item in colección]. Léela de izquierda a derecha: «construye esto, por cada elemento de aquello». Es más corta y muestra de un vistazo que todo es una lista nueva.
prices = [184.0, 185.0, 183.0]
# The loop way
notionals = []
for price in prices:
notionals.append(price * 100)
print(notionals) # [18400.0, 18500.0, 18300.0]
# The comprehension way: the same list in one line
notionals = [price * 100 for price in prices]
print(notionals) # [18400.0, 18500.0, 18300.0]Quedarse solo con algunos elementos: if
Añade if condition al final para quedarte solo con los elementos que pasan una prueba. Así se filtra: rendimientos positivos, órdenes por encima de un tamaño, símbolos que coinciden. Los elementos que no la pasan simplemente se omiten y se conserva el orden del resto.
# Keep only the items that pass a test
returns = [0.01, -0.02, 0.005, -0.001, 0.03]
gains = [r for r in returns if r > 0]
print(gains) # [0.01, 0.005, 0.03]Posiciones y pares
La parte for funciona como un bucle normal: puede recorrer range(...) para usar posiciones y nombrar las dos mitades de un par (for symbol, price in quotes). Usa posiciones cuando cada resultado depende de un vecino, como un rendimiento a partir del precio anterior; usa pares cuando cada elemento contiene varios campos.
# Simple returns from a price series, using positions
prices = [100.0, 102.0, 101.0]
returns = [prices[i] / prices[i - 1] - 1 for i in range(1, len(prices))]
print([round(r, 4) for r in returns]) # [0.02, -0.0098]
# Pairs can be unpacked, and the result can be transformed
quotes = [("aapl", 184.5), ("msft", 410.2), ("goog", 171.0)]
above = [symbol.upper() for symbol, price in quotes if price > 180]
print(above) # ['AAPL', 'MSFT']if/else va delante
Un if final elimina elementos. Si quieres cambiar cada elemento según una condición, usa la expresión condicional a if condition else b como primera parte. Confundir ambas es un error habitual: if solo va después del for, if ... else va antes.
# if/else changes each item; a plain if (at the end) drops items
returns = [0.01, -0.02, 0.005, -0.001, 0.03]
signals = ["long" if r > 0 else "flat" for r in returns]
print(signals) # ['long', 'flat', 'long', 'flat', 'long']for normal. Si una comprensión se vuelve larga o anidada, divídela en un bucle o en una función auxiliar: la legibilidad vale más que la astucia.[expression for item in items]: un resultado por elemento.... if conditional final: quedarse solo con algunos elementos.a if condition else bal principio: cambiar cada elemento.
A continuación: listas, la colección con la que han trabajado estos bucles y comprensiones: indexación, segmentación, añadir, quitar y ordenar.
Lab: qué se califica
- Valorar cada operación en una expresión con
notionals25 pts - Convertir precios en rendimientos con
simple_returns25 pts - Quedarse solo con las ganancias con
positive_returns25 pts - Filtrar y limpiar símbolos a la vez con
symbols_above25 pts