zip¿Con análogo en Python?

Question

¿Cuál es el análogo de la función zipWith de Haskell en Python?

zipWith :: (a -> b -> c) -> [a] -> [b] -> [c] 

Answer 1

Usted puede crear el suyo, si lo desea, pero en Python que en su mayoría hacer

list_c = [ f(a,b) for (a,b) in zip(list_a,list_b) ] 

como Python no es inherentemente funcional. Simplemente sucede que admite algunos modismos de conveniencia.

Answer 2

Puede usar el mapa:

>>> x = [1,2,3,4] 
>>> y = [4,3,2,1] 
>>> map(lambda a, b: a**b, x, y) 
[1, 8, 9, 4] 

Answer 3

map()

map(operator.add, [1, 2, 3], [3, 2, 1]) 

Aunque una LC con zip() se utiliza generalmente.

[x + y for (x, y) in zip([1, 2, 3], [3, 2, 1])] 

Answer 4

general como otros han mencionado mapa y postal puede ayudar a replicar la funcionalidad de zipWith como en Haskel.

Generalmente usted puede aplicar un operador binario definido o alguna función binaria en dos list.An ejemplo para reemplazar un zipWith Haskel con Python mapa/postal

Input: zipWith (+) [1,2,3] [3,2,1] 
Output: [4,4,4] 

>>> map(operator.add,[1,2,3],[4,3,2]) 
[5, 5, 5] 
>>> [operator.add(x,y) for x,y in zip([1,2,3],[4,3,2])] 
[5, 5, 5] 
>>> 

Hay otra variación de zipWith aka zipWith3, zipWith4 .... zipWith7. Para replicar estos funcionalistas es posible que desee utilizar izip y imap en lugar de zip y mapa.

>>> [x for x in itertools.imap(lambda x,y,z:x**2+y**2-z**2,[1,2,3,4],[5,6,7,8],[9,10,11,12])] 
>>> [x**2+y**2-z**2 for x,y,z in itertools.izip([1,2,3,4],[5,6,7,8],[9,10,11,12])] 
[-55, -60, -63, -64] 

Como puede ver, puede operar con cualquier cantidad de listas que desee y puede seguir utilizando el mismo procedimiento.

Answer 5

Un perezoso zipWith con itertools:

import itertools 

def zip_with(f, *coll): 
    return itertools.starmap(f, itertools.izip(*coll)) 

Esta versión generaliza el comportamiento de zipWith con cualquier número de iterables.

Answer 6

Sé que esto es una vieja pregunta, pero ...

Ya se ha dicho que la forma de pitón típica sería algo así como

results = [f(a, b) for a, b in zip(list1, list2)] 

y así ver una línea como la que, en su código , la mayoría de los pythonistas lo entenderán bien.

También ha ya sido un (creo) ejemplo puramente perezoso se muestra:

import itertools 

def zipWith(f, *args): 
    return itertools.starmap(f, itertools.izip(*args)) 

pero creo que starmap devuelve un iterador, por lo que no será capaz de índice, o pasar por varias veces lo esa función volverá.

Si no está particularmente preocupado por la pereza y/o la necesidad de índice o bucle a través de la nueva lista varias veces, esto es probablemente como propósito general como usted podría conseguir:

def zipWith(func, *lists): 
    return [func(*args) for args in zip(*lists)] 

No es que no pudiste No lo hagas con la versión perezosa, pero también podrías llamar a esa función como si ya hubieras creado tu lista de listas.

results = zipWith(func, *lists) 

o simplemente como normal como:

results = zipWith(func, list1, list2) 

De alguna manera, dicha llamada de función sólo se ve más simple y más fácil de asimilar que la versión lista por comprensión.

---

En cuanto a eso, esto parece extraña reminiscencia de otra función auxiliar A menudo escribo:

def transpose(matrix): 
    return zip(*matrix) 

que luego podría ser escrita como:

def transpose(matrix): 
    return zipWith(lambda *x: x, *matrix) 

No es realmente una versión mejor, pero siempre me parece interesante cómo cuando escribo funciones genéricas en un estilo funcional, a menudo me encuentro yendo, "Oh. Esa es solo una forma más general de una función que ya he escrito antes".