说明：

Question

我遇到了一个问题，涉及将List拆分成不同大小的块。我想要一个随机分成2或3对的列表。

示例：

L = [1,1,1,1,1,1,1,1,1,1,1,1]

我希望得到像：

L2 = [(1,1,1),(1,1),(1,1),(1,1,1),(1,1)]

但我希望这是随机的，这样每次代码运行时对和tripplet的分布都会发生变化。

有人可以帮忙吗？

Answer 1

作为一种更通用的方法，您可以使用以下功能：

from itertools import count
import random
def my_split(lst, chunks):
    def chunk_creator():
        total = 0
        while total <= len(lst):
            x = random.choice(chunks)
            yield L[total: x + total]
            total += x
        yield total - x

    def chunk_finder():
        for _ in count():
            chunk = list(chunk_creator())
            total = chunk.pop(-1)
            if total == len(L):
                return chunk[:-1]
    if max(chunks) <= len(L):
        return chunk_finder()
    else:
        return None

演示：

>>> L = [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1] 
>>> my_split(L, (2, 3))
... [[1, 1], [1, 1], [1, 1], [1, 1, 1], [1, 1, 1]]
>>> my_split(L, (2, 3))
... [[1, 1, 1], [1, 1, 1], [1, 1, 1], [1, 1, 1]]

说明：

此功能由子功能组成。第一个是chunk_creator，它的工作是根据列表的长度创建所需的块，并将它们作为迭代器返回。请注意，结束值是total变量，它是前面块的总和。

第二个函数（chunk_finder）将通过无限循环（itertools.count()）并检查total的值是否等于长度来找到我们所需的块输入列表。

Answer 2

有很多方法可以解决这个问题，也许你可以编写一个基于size函数返回大小的块的生成器：

import random

def take_chunked(i, size):
    idx = 0
    while idx < len(i):
        s = size(i[idx:])
        yield i[idx:idx+s]
        idx += s

def size_fun(i):
    if len(i) == 4:
        return 2
    if len(i) <= 3:
        return len(i)

    return random.randint(2,3)

输出：

>>> list(take_chunked("helloworld", size_fun))
['he', 'll', 'owo', 'rld']
>>> list(take_chunked("helloworld", size_fun))
['hel', 'low', 'or', 'ld']
>>> list(take_chunked("a", size_fun))
['a']
>>> list(take_chunked("", size_fun))
[]

只要列表中有足够的项目，此版本将保证块大小为2或3。

Answer 3

将len(L)分解为2s和3s的块，然后使用循环划分列表。

import numpy as np

def rand23():
    return np.random.randint(2,4)

def get_chunk(sz):
    rem = sz
    ch = []
    while ( rem > 0 ):
        if ( rem <= 3 ): #if <= 3 add what is left in the chunk (exit condition)
            ch.append(rem)
            rem = 0
            break
        elif ( rem == 4 ): #4 is the only edge case here
            ch.append(2)
            ch.append(2)
            rem = 0
            break
        else:
            ch.append(rand23())
            rem -= ch[-1]

    return ch

L = [1,1,1,1,1,1,1,1,1,1,1,1]
ch = get_chunk(len(L))
L2 = []
count = 0
#Divide the list into chunks based on ch
for c in ch:
    L2.append(tuple(L[count:count+c]))
    count += c

print L2

结果:(每行输出一次迭代）

[(1, 1, 1), (1, 1), (1, 1), (1, 1), (1, 1, 1)]

[(1, 1), (1, 1), (1, 1), (1, 1), (1, 1), (1, 1)]

[(1, 1, 1), (1, 1, 1), (1, 1, 1), (1, 1, 1)]

[(1, 1), (1, 1), (1, 1, 1), (1, 1, 1), (1, 1)]

[(1, 1, 1), (1, 1), (1, 1), (1, 1), (1, 1, 1)]

[(1, 1, 1), (1, 1, 1), (1, 1), (1, 1), (1, 1)]

[(1, 1), (1, 1), (1, 1, 1), (1, 1), (1, 1, 1)]

PS：你也可以递归地实现get_chunk()。

Answer 4

希望这有帮助。

步骤1：使用随机模块生成随机数

步骤2：使用随机数来决定它是对还是2（对于偶数随机数）/ 3（如果随机是偶数）。

第3步：编写代码。

from random import random

def selector():
    s = int(random() * 100 )
    return (s/2) == 0

L = [1 for i in range(30)]
L2 = []

while L:
    if selector():
        tmp = 2
    else:
        tmp = 3
    #print tmp
    if tmp > 0 and len(L) >= tmp:
        L2.append( [ L.pop() for i in range(tmp)] )
    if len(L) < 3:
        L2.append(set(L))
        L = []

print L, L2

Answer 5

这个很简单，可以解决问题： - ）

import random

L = [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1]
L2 = list()

i = 0
j = random.randint(2, 3)

while i < len(L):
    chunk = L[i:j]
    L2.append(chunk)
    i = j
    if len(L) - i == 4:  # case: 4 elements left in L
        j = i + 2
    elif len(L) - i < 4:  # case: 2 or 3 elements left in L
        j = len(L)
    else:
        j = i + random.randint(2, 3)

print L
print L2

Python：将列表分成不同大小的块

5 个答案:

说明：