python - 列表均匀分片示例代码

列表均匀分片示例代码：

"""
@author: zhangjun.xue
@time: 2020/6/30 16:32
@file: split_data_util.py
@desc: 对数据进行分片
"""
import traceback


def split_data(split_num, orgin_data):
    """
    :param split_num: 设置分片数
    :param orgin_data: 要分片的原始数据
    :return:
    """
    try:
        spli_data = []
        count = 0
        while count < split_num:
            spli_data.append([])
            count += 1
        for data in orgin_data:
            spli_data[orgin_data.index(data) % split_num].append(data)
        return spli_data
    except:
        raise Exception(traceback.format_exc())


if __name__ == "__main__":
    pass
    test_list = ["a", "b", "c", "d", "e", "f", "g", "h", "i", "j", "k"]

    # 分3片
    print(split_data(3, test_list))
    # 分5片
    print(split_data(5, test_list))

    # split_num = 0
    # split_count = 3
    # my_data_list_0 = []
    #
    # for index, data in enumerate(test_list):
    #     print('data = {} , index = {}'.format(data, index))
    #     if index % split_count == split_num:
    #         my_data_list_0.append(data)
    # print('my_data_list_0 = ', my_data_list_0)
    #
    # split_num = 1
    # my_data_list_1 = []
    # for index, data in enumerate(test_list):
    #     # print('data = {} , index = {}'.format(data, index))
    #     if index % split_count == split_num:
    #         my_data_list_1.append(data)
    # print('my_data_list_1 = ', my_data_list_1)
    #
    # split_num = 2
    # my_data_list_2 = []
    # for index, data in enumerate(test_list):
    #     # print('data = {} , index = {}'.format(data, index))
    #     if index % split_count == split_num:
    #         my_data_list_2.append(data)
    # print('my_data_list_2 = ', my_data_list_2)

输出：

[['a', 'd', 'g', 'j'], ['b', 'e', 'h', 'k'], ['c', 'f', 'i']]
[['a', 'f', 'k'], ['b', 'g'], ['c', 'h'], ['d', 'i'], ['e', 'j']]