pythonでfor in enumerate文で一部の要素が反応しない

Question

前提・実現したいこと

pythonで

def card(start,end):
  #省略

#使用例
card(0,3)

という関数を作り、
startからendまでの数で並び方が何通り在るか、その中でも0から始まるものとその数(あったら)を
タプル型で表示するようになるはずの関数を作っています。
card(0,2)だったら「0から始まるもの：(0,1,) (0,2,)
0以外の自然数で始まるもの：(1,0,)(1,2,)(2,0,)(2,1,)
4(+2)通り」という感じです。

発生している問題・エラーメッセージ

• 区別されるはずの0で始まるものが区別されない。

該当のソースコード

def card(start,end):
  import itertools
  li = list(itertools.permutations(range(start,end+1),end-start+1))
  print("=== 0から始まるもの ===")
  lix = 0
  for index, item in enumerate(li):
    if item[0] == 0:
      print(li[index])
      li.pop(index)
      lix += 1
  print("=== 普通のもの ===")
  print(li)
  print(str(len(li)) + "(+"+str(lix)+")通り")

#output

=== 0から始まるもの ===
(0, 1, 2, 3)
(0, 2, 1, 3)
(0, 3, 1, 2)
=== 普通のもの ===
[(0, 1, 3, 2), (0, 2, 3, 1), (0, 3, 2, 1), (1, 0, 2, 3), (1, 0, 3, 2), (1, 2, 0, 3), (1, 2, 3, 0), (1, 3, 0, 2), (1, 3, 2, 0), (2, 0, 1, 3), (2, 0, 3, 1), (2, 1, 0, 3), (2, 1, 3, 0), (2, 3, 0, 1), (2, 3, 1, 0), (3, 0, 1, 2), (3, 0, 2, 1), (3, 1, 0, 2), (3, 1, 2, 0), (3, 2, 0, 1), (3, 2, 1, 0)]
21(+3)通り

試したこと

いちどforループの中に何をチェックしているかprintで調べたところ、
(0, 1, 3, 2), (0, 2, 3, 1), (0, 3, 2, 1)のみ表示がされない=反応していませんでした。
処理に時間がかかると予想し、sleepで0.5秒止めてみたものの正常に作動は示せんでした。

補足情報（FW/ツールのバージョンなど）

replitで実行しています。
python v3.8.2

Answer 1

listに対してpopを使うと当然ながら要素の並びが変わります。
したがってlistをコピーした対象をループしたとしても、そのループから得られるindexはpop操作された後のlistでは意図した要素を指さない場合があります。

今回のやりたいことを実現するには、単純に０始まりまたはそれ以外のリストを作成するのがシンプルだと思います。

import itertools

l = list(range(10))
for i, v in enumerate(l.copy()):
    if v < 5:
        l.pop(i)
print(l) # [1, 3, 5, 7, 9]???


def card(start,end):
    l0, li = [], []
    for p in itertools.permutations(range(start,end+1),end-start+1):
        if p[0] == 0:
            l0.append(p)
        else:
            li.append(p)

    print("=== 0から始まるもの ===")
    print(*l0, sep='\n')
    print("=== 普通のもの ===")
    print(*li, sep='\n')
    print(f'{len(li)}+({len(l0)})通り')

card(0,3)
"""
=== 0から始まるもの ===
(0, 1, 2, 3)
(0, 1, 3, 2)
(0, 2, 1, 3)
(0, 2, 3, 1)
(0, 3, 1, 2)
(0, 3, 2, 1)
=== 普通のもの ===
(1, 0, 2, 3)
(1, 0, 3, 2)
(1, 2, 0, 3)
(1, 2, 3, 0)
(1, 3, 0, 2)
(1, 3, 2, 0)
(2, 0, 1, 3)
(2, 0, 3, 1)
(2, 1, 0, 3)
(2, 1, 3, 0)
(2, 3, 0, 1)
(2, 3, 1, 0)
(3, 0, 1, 2)
(3, 0, 2, 1)
(3, 1, 0, 2)
(3, 1, 2, 0)
(3, 2, 0, 1)
(3, 2, 1, 0)
18+(6)通り
"""

Comments

[me1t]

回答ありがとうございます！
本当に申し訳ありませんが、自己解決してしまいました！
また機会がありましたらよろしくおねがいします！

Answer 2

def card(start,end):
  import itertools
  li = list(itertools.permutations(range(start,end+1),end-start+1))
  print("=== 0から始まるもの ===")
  lix = 0
  liy = []
  for index, item in enumerate(li):
    if li[index][0] == 0:
      print(li[index])
      liy.append(li[index])
      lix += 1
  for item in enumerate(liy):
    li.remove(item[1])
  print("=== 普通のもの ===")
  for item in enumerate(li):
    print(item[1])
  print(str(len(li)) + "(+"+str(lix)+")通り")

・liの中の0からはじまるものの要素をlist「liy」に書き出す
・liの中からliyの中にあるものを消す+書き出す
・liの残ったものを書き出す

これで解決しました！

Answer 3

liを使うふぉっループの中でliを変更しているのでそうなります。

li.copy()とかli[:]を使ってコピーを取ってから回すようにしましょう。

def card(start,end):
  import itertools
  li = list(itertools.permutations(range(start,end+1),end-start+1))
  print("=== 0から始まるもの ===")
  lix = 0
  for index, item in enumerate(li.copy()):
    if item[0] == 0:
      print(li[index])
      li.pop(index)
      lix += 1
  print("=== 普通のもの ===")
  print(li)
  print(str(len(li)) + "(+"+str(lix)+")通り")

実行結果

>>> card(0,3)
=== 0から始まるもの ===
(0, 1, 2, 3)
(0, 2, 1, 3)
(0, 3, 1, 2)
(1, 0, 2, 3)
(1, 2, 0, 3)
(1, 3, 0, 2)
=== 普通のもの ===
[(0, 1, 3, 2), (0, 2, 3, 1), (0, 3, 2, 1), (1, 0, 3, 2), (1, 2, 3, 0), (1, 3, 2, 0), (2, 0, 1, 3), (2, 0, 3, 1), (2, 1, 0, 3), (2, 1, 3, 0), (2, 3, 0, 1), (2, 3, 1, 0), (3, 0, 1, 2), (3, 0, 2, 1), (3, 1, 0, 2), (3, 1, 2, 0), (3, 2, 0, 1), (3, 2, 1, 0)]
18(+6)通り

Comments

[me1t]

この場合だと
(1, 0, 2, 3)
(1, 2, 0, 3)
(1, 3, 0, 2)
が追加されてしまいます。
私は
(0, 1, 3, 2)
(0, 2, 3, 1)
(0, 3, 2, 1)
がついかしたいのです...