１行で表せるコードを分割して書きたい

Question

１行で表せるコードを分割して書きたいです。

# coding: utf-8
import itertools

a = [2,5,7,3,6]
b = [1,3,6,8,4]

print ( [ ( sum( [ dict( zip( a, b ) )[v] for i, v in enumerate(x) if i > a.index( v ) ] ), list(x) ) for x in itertools.permutations( a ) ] )

のprint文の一文でコードを分割して書きたいです。

今、

# coding: utf-8

a = [2,5,7,3,6]
b = [1,3,6,8,4]

def permutations(iterable, r=None):
    # permutations('ABCD', 2) --> AB AC AD BA BC BD CA CB CD DA DB DC
    # permutations(range(3)) --> 012 021 102 120 201 210
    pool = tuple(iterable)
    n = len(pool)
    r = n if r is None else r
    if r > n:
        return
    indices = list(range(n))
    cycles = list(range(n, n-r, -1))
    yield tuple(pool[i] for i in indices[:r])
    while n:
        for i in reversed(range(r)):
            cycles[i] -= 1
            if cycles[i] == 0:
                indices[i:] = indices[i+1:] + indices[i:i+1]
                cycles[i] = n - i
            else:
                j = cycles[i]
                indices[i], indices[-j] = indices[-j], indices[i]
                yield tuple(pool[i] for i in indices[:r])
                break
        else:
            return


for x in permutations(a):
   for i, v in enumerate(x):
     if i > a.index(v):
       print(sum([dict(zip(a, b))])[v])

で最初のコードとまったく同じ意味だと思ったのですが、

Traceback (most recent call last):
  File "test4.py", line 65, in <module>
    print(sum([dict(zip(a, b))])[v])
TypeError: unsupported operand type(s) for +: 'int' and 'dict'

とエラーが出ました。
何が間違っているのでしょうか？

Answer 1

以下のようになります

import itertools

a = [2,5,7,3,6]
b = [1,3,6,8,4]

result = []
for x in itertools.permutations(a):
    tmp = []
    for i, v in enumerate(x):
        if i > a.index(v):
            # 整数1, 3, 6, 8, 4 のどれかがtmpに格納されていく
            tmp.append(
                dict(zip(a, b))[v]
            )

    result.append(
        (sum(tmp), list(x))
    )
print(result)

リスト内包表記なんかを通常のループに分解する場合は、.append()等で要素を追加することを忘れないようにしておくと良いです。
今回の例は複雑ですが、以下のようにしてpprintで見やすく表示したり少しづつ式を分解していくと途中の経過がわかり、仕組みの理解につながります。

import itertools

a = [2, 5, 7, 3, 6]
b = [1, 3, 6, 8, 4]

from pprint import pprint
# もともとの
pprint([(sum([dict(zip(a, b))[v] for i, v in enumerate(x) if i > a.index(v)]), list(x)) for x in itertools.permutations(a)])

# sum部分をはずす
pprint([([dict(zip(a, b))[v] for i, v in enumerate(x) if i > a.index(v)], list(x)) for x in itertools.permutations(a)])

# [v]などの辞書キー部分をはずす
pprint([([dict(zip(a, b)) for i, v in enumerate(x) if i > a.index(v)], list(x)) for x in itertools.permutations(a)])

Answer 2

とりあえず、そのコードと元コードに適当に改行を入れて見直してみては？

sum(
  [
    dict(
      zip( a, b )
    )[v] for i, v in enumerate(x) if i > a.index( v )
  ]
)

sum(
  [
    dict(
      zip(a, b)
    )
  ]
)[v]

だいぶ違いますよね。

Comments

[退会済みユーザー]

回答ありがとうございます。私もそう考えて、
sum(
[
dict(
zip(a, b)
)
]
)[v]
の上に
for x in permutations(a):
for i, v in enumerate(x):
if i > a.index(v):
で条件分岐やfor文を持ってきたのにうまくいきませんでした。なぜなのでしょうか？

[Lhankor_Mhy]

後置ifの内包表記について、もう少し学ぶ必要があるように感じました。
http://qiita.com/y__sama/items/a2c458de97c4aa5a98e7
↑こちらを参照しては？