search
LoginSignup
1

posted at

updated at

Python リストをなるべく均等にN分割する

はじめに

マルチスレッド(スレッドプール等)で並行処理するときに使えるテクニックです

方法

import random

def split_array(ar, n_group):
    for i_chunk in range(n_group):
        yield ar[i_chunk * len(ar) // n_group:(i_chunk + 1) * len(ar) // n_group]

for _ in range(3):
    # 要素数をランダムに決定
    elem_count = random.randint(100,1000)
    ar = [i for i in range(elem_count)]
    # 分割数をランダムに決定
    n = random.randint(10,20)

    splited = split_array(ar, n)

    print(f'要素数{elem_count}のリストを{n}分割')
    for elem in splited:
        print(len(elem))
結果
要素数416のリストを12分割
34
35
35
34
35
35
34
35
35
34
35
35
要素数921のリストを13分割
70
71
71
71
71
71
70
71
71
71
71
71
71
要素数702のリストを10分割
70
70
70
70
71
70
70
70
70
71

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
What you can do with signing up
1