Как мне перебрать отношения в списке только один раз? - PullRequest
1 голос
/ 14 сентября 2009

У меня есть список пользователей:

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

Я хочу вычислить отношения между ними:

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

Как зациклить пользователей так, чтобы отношения между пользователями вычислялись только один раз?

Ответы [ 6 ]

11 голосов
/ 14 сентября 2009

Если вы заботитесь только об упорядоченных отношениях, вы можете сделать следующее:

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


1 2
2 3
3 4
4 5

если вам нужны все комбинации, вы должны использовать 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
0 голосов
/ 14 сентября 2009
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}

Используйте itertools.permuations вместо itertools.combission, если совместимость (a, b) не означает то же самое, что совместимость (b, a).

0 голосов
/ 14 сентября 2009

Если вы имеете в виду, что:

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

вы можете использовать:

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

это также вычислит совместимость между теми же пользователями (возможно, применимо)

0 голосов
/ 14 сентября 2009

Мне удалось сделать то, что я хотел с этим:

i = 0
for user1 in users:   
    i += 1     
    for user2 in users[i:]:
        print compatibility( user1, user2 )
0 голосов
/ 14 сентября 2009

Должно работать что-то вроде следующего (не проверено):

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 голосов
/ 14 сентября 2009

используется для циклов или для понимания списка.

Вот пример цикла:

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

Понимание списка:

score = [compatibility(x, y) for x in users for y in users if x!=y and compatibility(x,y) not in score]
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...