python リスト型 remove文

Question

data = [1,1,2,1,3,4]
for num in data:
 if data.count(num) < 2:
  data.remove(num)
return data

実行結果が[1,1,1,4]となるのはなぜでしょうか？
二個以上ある要素だけ残したつもりだっだのですが。。。
よろしくお願いします

Answer 1

ループ中のその要素を削除したりするとすべての要素分、ループできません。
data を複製したもの(data[:]) をforに指定すればどうでしょうか。

data = [1,1,2,1,3,4]
for num in data[:]:

Comments

[torano]

ループって内部ではそのリストのインデックス0→1→2→3→...の順に繰り返されるんですかね？
だからインデックス1の処理をしてる時に1を削除するとインデックス2が1に、3が2がに、とインデックスが繰り下げらるため、修正したい要素が二つ連続してるとうまくいかなかったのでしょうかね。

ありがとうございました

Answer 2

リファレンスにこの事象に関する記載があります。
7.3 for文 注釈

どの要素が次に使われるかを追跡するために、内部的なカウンタが使われており、このカウンタは反復処理を行うごとに加算されます。このカウンタがシーケンスの長さに達すると、ループは終了します。このことは、スイート中でシーケンスから現在の ( または以前の ) 要素を除去すると、 ( 次の要素のインデクスは、すでに取り扱った要素のインデクスになるために ) 次の要素が飛ばされることを意味します。

わかりにくいと思うので、コードも。

data = [0,1,2,3,4,5]
counter = 0

for num in data:
 print "--------------"
 print "counter       " + str(counter)
 print "before num    " + str(num)
 print "before length " + str(len(data))
 print "before list   " + str(data)
 if data.count(num) < 2:
  data.remove(num)
  print "--------------"
  print "aftre num     " + str(num)
  print "after length  " + str(len(data))
  print "after removed " + str(data)
 counter += 1
print "after process " + str(data)

実行結果

--------------
counter       0
before num    0
before length 6
before list   [0, 1, 2, 3, 4, 5]
--------------
aftre num     0
after length  5
after removed [1, 2, 3, 4, 5]
--------------
counter       1
before num    2
before length 5
before list   [1, 2, 3, 4, 5]
--------------
aftre num     2
after length  4
after removed [1, 3, 4, 5]
--------------
counter       2
before num    4
before length 4
before list   [1, 3, 4, 5]
--------------
aftre num     4
after length  3
after removed [1, 3, 5]
--------------
counter       3
after process [1, 3, 5]

本来であれば全ての要素が削除されるはずですが、
実際に削除されたのは、[0, 2, 4]のみとなっていて、
次の要素の処理がスキップされています。

対処としては、yoshi777さんがおっしゃっている様に、リストのコピーを使用しての
for文を実行するのが良いかと思います。

Comments

[torano]

理解が深まりました。ありがとうございます。

Answer 3

参考までに、こういうときはリスト内包表記やfilter等を使って新たにlistを生成した方がいい場合があります。

リスト内包表記

data = [1,1,2,1,3,4]
data = [i for i in data if data.count(i) < 2]
return data

filter ※Python2ではlistが返りますが、Python3ではイテレーターを返すので注意。

data = [1,1,2,1,3,4]
data = filter(lambda i: data.count(i) < 2, data)
return data

Comments

[torano]

そうだったのですね、ありがとうございます。