Python——按比例(随机)划分列表

                                                       Python——按比例(随机)划分列表

 

目标：对列表A按比例（随机）划分为子列表A1与A2。

代码：

import random


def data_split(full_list, ratio, shuffle=False):
    """
    数据集拆分: 将列表full_list按比例ratio（随机）划分为2个子列表sublist_1与sublist_2
    :param full_list: 数据列表
    :param ratio:     子列表1
    :param shuffle:   子列表2
    :return:
    """
    n_total = len(full_list)
    offset = int(n_total * ratio)
    if n_total == 0 or offset < 1:
        return [], full_list
    if shuffle:
        random.shuffle(full_list)
    sublist_1 = full_list[:offset]
    sublist_2 = full_list[offset:]
    return sublist_1, sublist_2


if __name__ == "__main__":
    data = list(range(20))
    sub_data1, sub_data2 = data_split(data, ratio=0.2, shuffle=True)
    print(sub_data1)
    print(sub_data2)

运行结果：

[9, 0, 17, 19]
[14, 18, 6, 16, 10, 8, 2, 4, 15, 3, 5, 7, 11, 13, 12, 1]