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']]