python代码比例_Python-scale（随机）列表,按,比例,划分

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]