何重にもネストされたfor文をPythonicに書く方法を探しています

Question

• 以下のようなコードは非常に保守性が低いです。もともとのデータ型がイケてないのは分かりますが、map関数などを使って少しでも読みやすく保守性を上げる方法はないでしょうか。

# sample data
pairList = [["米", "パン"], ["味噌汁", "コーンポタージュ"]]
dataJson = {
    "米": ["かに", "フランス"],
    "味噌汁": ["豚汁", "インディカ"],
    "パン": ["かに", "フランス"],
    "コーンポタージュ": ["ポタージュ", "ジャポニカ"],
}
# algorithm
mainDishList = []
for i in pairList:
    if i[0] in dataJson.keys() and i[1] in dataJson.keys():
　　　　# この辺りをもう少し組み込み関数などでうまく書きたい
        for x in dataJson[i[0]]:
            for y in dataJson[i[1]]:
                if x[0] == y[0] and x[1] == y[1]:
                    print("ok")

Comments

[quickquip]

なにがしたかったか当てるクイズになってますね。

Answer 1

setを使ってみました。
要件がいまいち不明ですが、pairの要素数、dataJsonの値のリストの要素すによらず判定できます。

# sample data
pairList = [["米", "パン"], ["味噌汁", "コーンポタージュ"]]
dataJson = {
    "米": ["かに", "フランス"],
    "味噌汁": ["豚汁", "インディカ"],
    "パン": ["かに", "フランス"],
    "コーンポタージュ": ["ポタージュ", "ジャポニカ"],
}

keys = set(dataJson.keys())

for pair in pairList:
    print(pair)
    if not set(pair) - keys: # すべてがdataJsonのキーとして存在
        # キーに対応する各値のリストがすべて同一か
        values = set(tuple(dataJson[k]) for k in pair)
        if len(values) == 1:
            print('ok')

Answer 2

何をしたいのかがよくわからないコードです。
ちょっと追加して実行してみました。

>>> pairList = [["米", "パン"], ["味噌汁", "コーンポタージュ"]]
>>> dataJson = {
...     "米": ["かに", "フランス"],
...     "味噌汁": ["豚汁", "インディカ"],
...     "パン": ["かに", "フラメンコ"],
...     "コーンポタージュ": ["ポタージュ", "ジャポニカ"],
... }
>>>
>>> for i in pairList:
...     if i[0] in dataJson.keys() and i[1] in dataJson.keys():
...         for x in dataJson[i[0]]:
...             for y in dataJson[i[1]]:
...                 if x[0] == y[0] and x[1] == y[1]:
...                     print(i, x[0], x[1])
...                     print("ok")
...
['米', 'パン'] か に
ok
['米', 'パン'] フ ラ
ok

米はフランスで、パンはフラメンコで違っていますが、OKとなっています。こういう結果を欲しいのでしょうか？

たぶんこうだったのだろうコードを書いてみました。

>>> pairList = [["米", "パン"], ["味噌汁", "コーンポタージュ"]]
>>> dataJson = {
...     "米": ["かに", "フランス"],
...     "味噌汁": ["豚汁", "インディカ"],
...     "パン": ["かに", "フランス"],
...     "コーンポタージュ": ["ポタージュ", "ジャポニカ"],
... }
>>>
>>> for first, second in pairList:
...     try:
...         if dataJson[first] == dataJson[second]:
...             print(first, second, dataJson[first])
...             print('ok')
...     except KeyError:
...         pass
...
米 パン ['かに', 'フランス']
ok

欲しいものがこれでないのであれば、それを日本語で説明してください。

Answer 3

何重にもネストされたfor文をPythonicに書く方法を探しています

itertools.productを使うと多重forループを単一forループに統合できます。

import itertools

for x,y in itertools.product(dataJson[i[0]], dataJson[i[1]]):
   pass # x,y を使った処理