Tengo una lista de usuarios:

users = [1,2,3,4,5]

Quiero calcular una relación entre ellas:

score = compatibility( user[0], user[1] )

¿Cómo hago un bucle sobre los usuarios para que una relación entre usuarios se calcule solo una vez?

1
Hobhouse 14 sep. 2009 a las 14:19

6 respuestas

La mejor respuesta

Si solo le importa la relación ordenada, puede hacer lo siguiente:

>>> for i, u in enumerate(users[1:]):
    print(users[i], u)           # or do something else


1 2
2 3
3 4
4 5

Si necesita todas las combinaciones, debe usar itertools.combinations:

>>> import itertools
>>> for i in itertools.combinations(users, 2):
    print(*i)

1 2
1 3
1 4
1 5
2 3
2 4
2 5
3 4
3 5
4 5
11
SilentGhost 14 sep. 2009 a las 17:51

Utilizar para bucles, o lista de comprensión.

Aquí hay un ejemplo de bucle:

for u in users:
    for su in users:
        if su == u:
            pass
        else:
            score = compatibility(u, su)
            # do score whatever you want

Lista de comprensión:

score = [compatibility(x, y) for x in users for y in users if x!=y and compatibility(x,y) not in score]
0
Mohamed 14 sep. 2009 a las 10:43
import itertools

def compatibility(u1, u2):
    "just a stub for demonstration purposes"
    return abs(u1 - u2)

def compatibility_map(users):
    return dict(((u1, u2), compatibility(u1, u2))
         for u1, u2 in itertools.combinations(users, 2))

> compat.compatiblity_map([1,2,3,4,5])
{(1, 2): 1, (1, 3): 2, (4, 5): 1, (1, 4): 3, (1, 5): 4,
 (2, 3): 1, (2, 5): 3, (3, 4): 1, (2, 4): 2, (3, 5): 2}

Utilice itertools.permuations en lugar de itertools.combinations si la compatibilidad (a, b) no significa lo mismo que la compatibilidad (b, a).

0
bendin 14 sep. 2009 a las 10:40

Si quieres decir eso:

compatibility(user[0], user[1]) == compatibility(user[1], user[0])

Podrías usar:

for i, user1 in enumerate(users):
    for user2 in users[i:]:
        score = compatibility(user1, user2)

Esto también calculará la compatibilidad entre los mismos usuarios (tal vez aplicable)

0
Wim Verhavert 14 sep. 2009 a las 10:39

Logré hacer lo que quería con esto:

i = 0
for user1 in users:   
    i += 1     
    for user2 in users[i:]:
        print compatibility( user1, user2 )
0
Hobhouse 14 sep. 2009 a las 10:31

Algo como lo siguiente debería funcionar (no probado):

users_range = range(len(users))

# Initialize a 2-dimensional array
scores = [None for j in users_range for i in users_range]

# Assign a compatibility to each pair of users.
for i in users_range:
    for j in users_range:
        scores[i][j] = compatibility(users[i], users[j])
0
Fragsworth 14 sep. 2009 a las 10:30