将阵列分成不均匀的组

Question

在python中给出这样的数组：

a = [ 0, 1, 3, 4, 6, 7, 8, 10, 14 ]

我想将其分为三个不平衡的组，以便最终得到这样的结果：

b = [0, 1, 3, 4]

c = [6, 7, 8]

d = [10, 14]

我想将数字分组为5的倍数，因此0-4的任何整数都将以第一个数组结尾，第二个数组以5-9结尾，依此类推。

Answer 1

Itertools.groupby始终是答案！在这里，我们将每个数字四舍五入到最接近的5，然后按相等的数字分组：>>> for n, g in itertools.groupby(a, lambda x: round(x/5)*5):    print list(g)[0, 1, 3, 4][6, 7, 8][10, 14]

Answer 2

如果我们对正在使用的数字有所了解，我们可能会或多或少地节省时间。我们还可以想出一个非常快速的方法，它的内存效率非常低，但是如果适合您的目的，可以考虑一下：#something to store our new lists inrange = 5 #you said bounds of 5, right?s = [ [] ]for number in a:    foundit = false    for list in s:        #deal with first number        if len( list ) == 0:            list.append( number )        else:            #if our number is within the same range as the other number, add it            if list[0] / range == number / range:                foundit = true                list.append( number )    if foundit == false:       s.append( [ number ] )

Answer 3

现在，我对您对组的定义有了更好的了解，我认为这个相对简单的答案不仅会起作用，而且还应该非常快：from collections import defaultdicta = [0, 1, 3, 4, 6, 7, 8, 10, 14]chunk_size = 5buckets = defaultdict(list)for n in a:    buckets[n/chunk_size].append(n)for bucket,values in sorted(buckets.iteritems()):    print '{}: {}'.format(bucket, values)输出：0: [0, 1, 3, 4]1: [6, 7, 8]2: [10, 14]