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 in
range = 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 defaultdict

a = [0, 1, 3, 4, 6, 7, 8, 10, 14]
chunk_size = 5
buckets = 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]

将阵列分成不均匀的组

3 个答案: