Python：リスト内包表記の中で二重ループと条件分岐をする

Question

リスト内包表記の中で二重のループとif文での条件分岐をしたいのですが、わかりません。

一番基本的な内包表記の書き方については理解できていると思うのですが、for文が二回出てきたりするといまいちわからなくなってしまうレベルです。

lst = [1, 2, 3, 4]
lst2 = [4, 5, 6, 2]

for x in lst:
    for y in lst2:
            if x == y:
                   lst.remove(x)

現在、上記のようなコードを考えています。
二つのリストがあり、それらの要素を比較して等しいものがあればlstから除去する、という処理を想定しています。

pythonの内包表記を少し詳しくの下の方に、内包表記で二重のfor文についての記述があるのですが、いまいちわかりませんでした。
サイトの内包表記はなんとか理解できるのですが、上記のように異なるリストの要素同士を比較して同じなら削除する、という処理にする場合はどのように書けばいいのでしょうか？
また、そもそも上記のような処理の場合は内包表記にせず、for文を二回書いてif文で判定するような処理の方が良いのでしょうか？

ご教示いただけますと幸いです。よろしくお願いいたします。

Answer 1

for文でlstの要素を順番に参照しながら、そのfor文の中でlstの要素をremove(or append)するとバグるのでやめたほうがいいですよ。

forループ使うなら

lst = [1, 2, 3, 4]
lst2 = [4, 5, 6, 2]

lst_new = lst[:]  # make copy
for x in lst:
    if x in lst2:
        lst_new.remove(x)

リスト内法表記使うなら

lst = [1, 2, 3, 4]
lst2 = [4, 5, 6, 2]

lst = [x for x in lst if x not in lst2]

Comments

[Ykkykk]

ご回答いただきありがとうございます。
そもももremoveを使うとバグが起こるのですね。appendの場合もバグが起こるということですよね？？初めて知りました。
ありがとうございました！

[tachikoma]

試しに下記を実行してみてください。これだけで無限ループになります。（そのうちメモリーが枯渇するのでctr+cとかで止めてくださいね）
lst = [1]
for x in lst: lst.append(x)

[Ykkykk]

実際にやってみるとよくわかりました。
例までお教えいただきありがとうございます！

Answer 2

そもそも二重のfor文にする必要が無いです。
また、内包表記を書く際は要素を増やしていくことに主眼を置きます。

lst = [1, 2, 3, 4]
lst2 = [4, 5, 6, 2]

dst = []
for x in lst:
    if x not in lst2:
        dst.append(x)

lst = dst

これを踏まえて内包表記化します。

lst = [
    x for x in lst
    if x not in lst2
]

Comments

[Ykkykk]

ご回答いただきありがとうございます。そもそもnot inを使用すれば二重のfor文にする必要がないのですね。
また、内包表記は「要素を増やしていくことに主眼を置」くということがわかりました。
ありがとうございました！