itertoolsからintで要素をもらう方法

Question

itertoolsのデカルト積（？…という名前も初めて知りました；；）で配列にもらったものを合計したいのですが、配列内にstrで入ってくるため、エラーとなってしまいます。

for element in itertools.product('972', repeat=N):
	print (sum(int(element)))

何か小細工を弄すればできるんじゃないかと、上記のように書いてみましたが、やはりダメでした。
配列をループで回して、int変換して合計すればできると思うのですが、itertoolsがそもそもintで配列を作ってくれればsumでできると思うので調べているのですが、現時点で解決できていません。
がんばってループで回す方法しかないものでしょうか？

Answer 1

質問に書いてあるとおり、毎ループでintすることが無駄なので最初に変換するべきで、

for element in itertools.product(map(int, '972'), repeat=N):
    print(sum(element))

がいいでしょうね。

Comments

[yukiko]

うわー、すみません。最初の回答の方にお返事しているまに、回答頂いていたんですね。
magichanさんと同じ回答なので、きっとポピュラーな書き方なんですね。
python使いの方には常識…なのかもですね。スミマセン。
ありがとうございます。

[quickquip]

なにも問題はありませんよ。
素直な問題なので答えが同じになってもおかしくないでしょうね。

Answer 2

こういうこと？

for element in itertools.product(map(int, '972'), repeat=N):
    print(sum(element))

Comments

[magichan]

おっ！
完全に quiquiさんとかぶってしまいました
申し訳ありません。

[yukiko]

はい、そういうことでした＾＾
うわー、すみません。最初の回答の方にお返事しているまに、回答頂いていたんですね。
こちらも今実行してみました。シンプルに書けますね。
ありがとうございます。

Answer 3

elementをリスト内包表記でintリストに変換してみました。

import itertools
N = 3
for element in itertools.product('972', repeat=N):
    #l = [int(i) for i in element]
    #s = sum(l)
    #print(l)
    #print(s)
    print( sum([int(i) for i in element]))

Comments

[yukiko]

うわ…！！！すごいです。あっというまに回答が…。
今、実行してみたら、ちゃんと動きました。
リスト内包表記というんですね。
itertoolsに何かのパラメータを与えてintにする方法があるのかなあと思っていました。
助かります。勉強します。ありがとうございます。

[can110]

リスト内包表記は分かりやすく便利なのですが今回のケースでは無駄があるので
quiquiさん、magichanさんの回答の方が適切かなあと思います。