Некрасивая комбинация выражения генератора с циклом for - PullRequest
8 голосов
/ 01 марта 2010

В моем коде Python 2.6 появляется следующее:

for src, dst in ([s,d] for s in universe for d in universe if s != d):

Могу ли я сделать намного лучше? Что мне особенно не нравится, так это то, что я задаю одну и ту же пару дважды, один раз для цикла for и снова для выражения генератора. Я не уверен, предпочел бы я:

for src, dst in itertools.product(universe, universe):
    if src != dst:

Есть ли способ выразить этот цикл кратко?

universe оказывается списком, если он имеет какое-либо значение. Порядок итерации не имеет значения.

Ответы [ 3 ]

5 голосов
/ 01 марта 2010

Вы можете использовать простые вложенные циклы for:

for src in universe:
   for dst in universe:
      if src == dst:
         continue
      ...

Я бы сказал, что в данном случае это самый простой для чтения синтаксис.

3 голосов
/ 01 марта 2010

Я предлагаю сохранить его полностью функциональным или полностью понятным. Вот реализация, которая полностью функциональна.

import itertools 
import operator

def inner_product(iterable):
    "the product of an iterable with itself"
    return itertools.product(iterable, repeat=2)

def same(pair):
    "does this pair contain two of the same thing?"
    return operator.is_(*pair)

universe = 'abcd'

pairs = inner_product(universe)
unique_pairs = itertools.ifilterfalse(same, pairs)
for pair in unique_pairs:
    print pair

"""
('a', 'b')
('a', 'c')
('a', 'd')
('b', 'a')
('b', 'c')
('b', 'd')
('c', 'a')
('c', 'b')
('c', 'd')
('d', 'a')
('d', 'b')
('d', 'c')
"""
1 голос
/ 01 марта 2010

itertools.product может принимать аргумент ключевого слова «repeat», если вы хотите иметь ту же последовательность, что и несколько параметров:

itertools.product(universe, repeat=2)

Вопрос в том, является ли это более читабельным.

Вы можете заменить исходный код на:

* +1007 *
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...