python把列表片分成n份_Python把列表中的数字尽量等分成n份

  • Post author:
  • Post category:python


问题描述:假设一个列表中含有若干整数,现在要求将其分成n个子列表,并使得各个子列表中的整数之和尽可能接近。

下面的代码并没有使用算法,而是直接将原始列表分成n个子列表,然后再不断地调整各个子列表中的数字,从元素之和最大的子列表中拿出最小的元素放到元素之核最小的子列表中,重复这个过程,知道n个子列表足够接近为止。

import random

def numberSplit(lst, n, threshold):

”’lst为原始列表,内含若干整数,n为拟分份数

threshold为各子列表元素之和的最大差值”’

length = len(lst)

p = length // n

#尽量把原来的lst列表中的数字等分成n份

partitions = []

for i in range(n-1):

partitions.append(lst[i*p:i*p+p])

else:

partitions.append(lst[i*p+p:])

print(‘初始分组结果:’, partitions)

#不停地调整各个子列表中的数字

#直到n个子列表中数字之和尽量相等

times = 0

while times < 1000:

times += 1

#元素之和最大的子列表和最小的子列表

maxLst = max(partitions, key=sum)

minLst = min(partitions, key=sum)