0


Python把列表中的数字尽量等分成n份

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

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

import random

def numberSplit(lst, n, threshold):

  1. '''lst为原始列表,内含若干整数,n为拟分份数
  2. threshold为各子列表元素之和的最大差值'''
  3. length = len(lst)
  4. p = length // n
  5. #尽量把原来的lst列表中的数字等分成n份
  6. partitions = []
  7. for i in range(n-1):
  8. partitions.append(lst[i*p:i*p+p])
  9. else:
  10. partitions.append(lst[i*p+p:])
  11. print('初始分组结果:', partitions)
  12. #不停地调整各个子列表中的数字
  13. #直到n个子列表中数字之和尽量相等
  14. times = 0
  15. while times < 1000:
  16. times += 1
  17. #元素之和最大的子列表和最小的子列表
  18. maxLst = max(partitions, key=sum)
  19. minLst = min(partitions, key=sum)
  20. #把大的子列表中最小的元素调整到小的子列表中
  21. m = min(maxLst)
  22. i = [j for j, v in enumerate(maxLst) if v==m][0]
  23. minLst.insert(0, maxLst.pop(i))
  24. print('第{0}步处理结果:'.format(times), partitions)
  25. first = sum(partitions[0])
  26. for item in partitions[1:]:
  27. if abs(sum(item)-first) > threshold:
  28. break
  29. else:
  30. break
  31. else:
  32. print('很抱歉,我无能为力,只能给出这样一个结果了。')
  33. return partitions

lst = [random.randint(1, 100) for i in range(10)]

print(lst)

result = numberSplit(lst, 3, 5)

print('最终结果:', result)

#输出各组数字之和

print('各子列表元素之和:')

for item in result:

  1. print(sum(item))

运行结果一:

运行结果二:

运行结果三:


本文转载自: https://blog.csdn.net/dongfuguo/article/details/118704749
版权归原作者 dongfuguo 所有, 如有侵权,请联系我们删除。

“Python把列表中的数字尽量等分成n份”的评论:

还没有评论