リストの要素の置換の方法がわかりません。

Question

前提・実現したいこと

Pythonを使用しています。

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

という2つのリストがあり、

a[i][0] == 1　に該当するaの要素（[1, 4][1, 6][1, 8]）を
bのリストに置換して

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

というリストを作りたいのですが、どうにもうまくいきません。
どうすればよいでしょうか。

試したこと

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

a_l = []
for i in range(len(a)):
if a[i][0] == 1:
a_l.append(a[i])
print(a_l)
Out[]: [[1, 4], [1, 6], [1, 8]]

ここまではなんとかなりましたが、その後がうまくいきません。

Answer 1

これで合っていますか？

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

target = {y for x,y in a if x == 1}
result = [[1, x[1]] if x[1] in target else x  for x in b]
print(result)  # => [[0, 0], [0, 1], [0, 2], [0, 3], [1, 4], [0, 5], [1, 6], [0, 7], [1, 8]]

Comments

[TeRa_130]

とてもわかりやすいです。
ありがとうございます。

Answer 2

bが0から連番でデータを保持している前提なら。

for e1, e2 in a:
    b[e2][0] = e1    # あるいは b[e2][0] = (e1 == 1)

print(b)

実行結果 Wandbox

[[0, 0], [0, 1], [0, 2], [0, 3], [1, 4], [0, 5], [1, 6], [0, 7], [1, 8]]

コードの書き方について

teratailには、コードを見やすく表示する機能があります。
質問編集画面を開き、コードを選択した状態で<code>ボタンを押してください。

特にPythonの場合、インデントが崩れるとコードの意味が変わってしまいます。

Comments

[TeRa_130]

コードの書き方についてまで教えていただき、
ありがとうございます。

Answer 3

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

al = list(filter(lambda elem:elem[0]==1, a))

for i in range(len(al)):
  b = [al[i] if elem[1] == al[i][1] else elem for elem in b]

print(list(al))
print(list(b))

#[[1, 4], [1, 6], [1, 8]]
#[[0, 0], [0, 1], [0, 2], [0, 3], [1, 4], [0, 5], [1, 6], [0, 7], [1, 8]]

これでいかがですか

Comments

[TeRa_130]

とてもわかりやすいです。
ありがとうございます。

Answer 4

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

d = {y: [x, y] for x, y in a if x == 1}
[d[z[1]] if z[1] in d else z for z in b]

d として、変換表をつくいります。
あとは、 b の各要素に対して、変換表を引いてヒットすれば変換結に置き換え、 ヒットしなければ、そのままの値 というようにしていきます。

実行例

Answer 5

for i in a_l:
    for j in b:
        if i[1]==j[1]:
            b[i[1]]=i         

汚いし、もっといい方法あるかもですがとりあえずこれでなんとかなると思います。

Comments

[TeRa_130]

やりたいことはこんな感じでした。
ありがとうございます。