nで割り切れる数の時に処理を行うPythonのコード

Question

前提・実現したいこと

• 実装そのものは出来ているのですが、コードがどうにも不格好です。Pythonらしく書き直す場合はどうすればよいでしょうか。

該当のソースコード

valueList = []
rangeList = list(range(1, 10))
print(rangeList)
for flag, i in enumerate(rangeList):
    valueList.append(i)
    if (flag + 1) % 4 == 0:
        print(f"flag is {flag}")
        print(f"value is {valueList}")
        flag = 0
        valueList = []
    if flag == len(rangeList) - 1:
        print(f"flag is {flag}")
        print(f"value is {valueList}")
        flag = 0
        valueList = []

出力

[1, 2, 3, 4, 5, 6, 7, 8, 9]
flag is 3
value is [1, 2, 3, 4]
flag is 7
value is [5, 6, 7, 8]
flag is 8
value is [9]

Comments

[y_waiwai]

なにをどうしたいんでしょうか。
問題がわかりません

[退会済みユーザー]

他の人には伝わったのですが、y_waiwaiさんには伝わらなかったようです。

Answer 1

やりたいことがリストを４個ずつに分割するのであれば以下が参考になります。
How do you split a list into evenly sized chunks?

def chunks(lst, n):
    """Yield successive n-sized chunks from lst."""
    for i in range(0, len(lst), n):
        yield lst[i:i + n]

rangeList = list(range(1, 10))

ret = list(chunks(rangeList, 4))
print(ret) # [[1, 2, 3, 4], [5, 6, 7, 8], [9]]

Comments

[退会済みユーザー]

求めていたものです。yieldを使うのですね、勉強になりました。

Answer 2

• Pythonらしく書き直す

十分Pythonらしいので、特に書き直す必要はないでしょう。
ただし、コードの中に10とか4とかいう数字を埋め込むのは止めた方が良いと思います。
そういうものは変数にしておきましょう。

>>> N = 4
>>> L = 10
>>> rangeList = list(range(1, L))
>>> for flag in range(0, L, N):
...     print(f"flag is {flag}\nvalue is {rangeList[flag:flag+N]}")
...
flag is 0
value is [1, 2, 3, 4]
flag is 4
value is [5, 6, 7, 8]
flag is 8
value is [9]
>>> 
>>> N = 7
>>> L = 19
>>> rangeList = list(range(1, L))
>>> for flag in range(0, L, N):
...     print(f"flag is {flag}\nvalue is {rangeList[flag:flag+N]}")
...
flag is 0
value is [1, 2, 3, 4, 5, 6, 7]
flag is 7
value is [8, 9, 10, 11, 12, 13, 14]
flag is 14
value is [15, 16, 17, 18]

Comments

[退会済みユーザー]

あくまでコードサンプルなので、実際のモジュールを張ると冗長になるかと思い、そのあたりは適当でした。
しかしコードは参考になりました。ありがとうございます。

Answer 3

https://stackoverflow.com/questions/312443/how-do-you-split-a-list-into-evenly-sized-chunks