Python：内包表記を用いたlistの書き換え

Question

前提・実現したいこと

Pythonを使用しています。

• 実現したい内容 …

「　listを内包表記を用いてb[i][1]の中にa[i][1]の情報があった時、
b[i][0]の情報をa[i][0]の情報に書き換えたいのですがうまくいきません。

これを解決したいです。」

実行したコードとその結果

実行したコード

このようなコードを実行しました。

In[]:
a = [['u', 0], ['u', 2], ['u', 4], ['u', 6], ['u', 8]]
b = [['n', 1], ['n', 2], ['n', 3], ['n', 4], ['n', 5]]

target = {y1 for x1,y1 in a if x1 !=0 }
way_li = [[x1[0], x1[1]] if x1[1] in target else x1  for x1 in b]

way_li

実行した結果

上記のコードを実行した結果、
以下のようになりました。

Out[]:
[['n', 1], ['n', 2], ['n', 3], ['n', 4], ['n', 5]]

本来得たい結果

望んでいた実行結果は以下の通りです。

Out[]:
[['n', 1], ['u', 2], ['n', 3], ['u', 4], ['n', 5]]

Answer 1

冷静に考えてみてもらえるとわかるとおり、

way_li = [[x1[0], x1[1]] if x1[1] in target else x1  for x1 in b]

の行で[x1[0], x1[1]]にしろx1にしろ同じものです。

[x1[0], x1[1]]と書く代わりに手打ちで["u", x1[1]]とするのが一つの手。でも、これだと要件を厳密に満たさないので（違うフォーマットのデータに対応できない）、その辺りを勘案すると、こうなりますかね。

a = [['u', 0], ['u', 2], ['u', 4], ['u', 6], ['u', 8]]
b = [['n', 1], ['n', 2], ['n', 3], ['n', 4], ['n', 5]]

target = {y:[x, y] for x, y in a}  # 辞書にして保存
way_li = [target.get(y, [x, y]) for x, y in b] # 辞書のgetメソッドでキーの有無を判定し、あれば置き換える

print(way_li)

Comments

[TeRa_130]

違うフォーマットの対応まで対応いただき、
ありがとうございます。

Answer 2

註：
この回答は、『b[i][1]の中にa[i][1]の情報があった時』という条件を素直に満たすものです。
ただし既にご指摘があるように、質問者さんの望む処理では無いようです。

aとbをzipを使って並行して巡回し、
条件を満たすときはaの要素を、そうでないときはbの要素を選択すれば良いでしょう。

way_li = [
    ea if ea[1] == eb[1] else eb
    for ea, eb in zip(a, b)
]

これは次のように書いているのと同じロジックです。

way_li = []
for ea, eb in zip(a, b):
    if ea[1] == eb[1]:
        way_li.append(ea)
    else:
        way_li.append(eb)

Comments

[hayataka2049]

インデックス3の要素を['u', 4]にしたいらしいので、これではないのだと思います。私もこっちで回答しかけてから気づきましたけど。

[LouiS0616]

ほんとですね。気付きませんでした。
ご指摘ありがとうございます。

[TeRa_130]

ありがとうございます。
条件分岐をすればわざわざ2つのコードを行き来しなくて済みますね。

[LouiS0616]

読み違えた身でケチを付けるのも恐縮ですが、
『b[i][1]がa[～][1]に含まれるとき』とか『b[i][1]がa[k][1]に含まれるとき』というような書き方だったら意図が伝わり易かったんじゃないかなぁ、と思います。

[TeRa_130]

確かに私の書き方も悪かったかもしれません。
ご指摘ありがとうございます。