if or文が機能しない

ごくごく簡単なif or 文に基づき、リストの要素を選定したいのですが、
なぜか動いてくれません。

【書いたコード】

python
1ls = [[2, 0], [2, -1], [4, 0], [2, 1], [1, 0]]
2for i in ls:
3    if i[0] < 0 or i[0] >= 3 or i[1] < 0 or i[1] >= 5:
4        ls.remove(i)
5
6print(ls)

【期待する出力】

python
1[[2, 0], [2, 1], [1, 0]] #if文により[2,-1]と[4,0]に消えてほしい

【実際の出力】

python
1[[2, 0], [4, 0], [2, 1], [1, 0]]

[4,0]が消えません。
何か演算子を間違えてるとか、スペースとかの問題だったりするんでしょうか。
色々ネットでif or文を調べたり、orを使わずに全部ifの入れ子にしてみたりしましたがダメでした。

すみませんがご教示いただければ助かります。

行動規範の内容に同意します

回答2件

ベストアンサー

巡回中のリストに要素を追加したり、削除したりしてはいけません。
次のように浅いコピーを取るのが簡単です。

Python
1ls = [[2, 0], [2, -1], [4, 0], [2, 1], [1, 0]]
2for i in ls[:]:
3    if i[0] < 0 or i[0] >= 3 or i[1] < 0 or i[1] >= 5:
4        ls.remove(i)
5
6print(ls)

実行結果 Wandbox

[[2, 0], [2, 1], [1, 0]]

元のコードがうまくいかない理由

実際に試してみると面白いです。

Python
1#
2lst = [1, 2, 3, 4, 5]
3for elem in lst:
4    print(elem)
5    lst.remove(elem)
6
7print(lst)
8
9#
10lst = [1, 2, 3, 4, 5]
11for elem in lst[:]:
12    print(elem)
13    lst.remove(elem)
14
15print(lst)

実行結果 Wandbox

ただし

今回は『排除する条件』より『排除しない条件』の方が書きやすそうです。
ついでにリスト内包表記も利用してみます。

Python
1ls = [[2, 0], [2, -1], [4, 0], [2, 1], [1, 0]]
2ls = [
3    row for row in ls
4    if 0 <= row[0] < 3
5    if 0 <= row[1] < 5
6]
7
8print(ls)

投稿2018/06/01 14:02

総合スコア35660

2018/06/01 14:07

LouiS0616さん、いつも早急にご回答いただきありがとうございます。浅いコピーを参照先にして、そこでひっかかった要素をコピー元のlsから抜く感じですね。 [:]が浅いコピーになるのは知りませんでした。元のやり方でできない理由がちょっとわからないですが、参考になりました、ありがとうございます。

LouiS0616

2018/06/01 14:11

> 元のやり方でできない理由がちょっとわからない追記しておきました。

Sigma1630

2018/06/01 14:18

あー！わかりました。ls[0],ls[1],ls[2]…って順番に消していくと頭から消した場合結果的に1個飛ばしになってしまう感じですね！ありがとうございます。内包表記もそろそろ覚えます。

行動規範の内容に同意します

for文で回しているリストを直接操作すると、理解しがたい現象が起こります。

python
1ls = [[2, 0], [2, -1], [4, 0], [2, 1], [1, 0]]
2for i in ls:
3    print(i)
4    if i[0] < 0 or i[0] >= 3 or i[1] < 0 or i[1] >= 5:
5        ls.remove(i)
6
7print(ls)
8""" =>
9[2, 0]
10[2, -1]
11[2, 1]
12[1, 0]
13[[2, 0], [4, 0], [2, 1], [1, 0]]
14"""

これは仕様なので（ドキュメントに書いてあるのかしら・・・）、このままのコードで行くのは無理です。

回避策は、

python
1ls = [[2, 0], [2, -1], [4, 0], [2, 1], [1, 0]]
2for i in ls[:]:  # 範囲指定なしのスライスでリストのコピーが作れる。本体とは違うリストを回す
3    if i[0] < 0 or i[0] >= 3 or i[1] < 0 or i[1] >= 5:
4        ls.remove(i)
5
6print(ls)

とか

python
1ls = [[2, 0], [2, -1], [4, 0], [2, 1], [1, 0]]
2ls = [i
3      for i in ls 
4      if not (i[0] < 0 or i[0] >= 3 or i[1] < 0 or i[1] >= 5)]  # 内包表記で元のリストは変更せず操作し、作ってから最後に再代入する
5print(ls)