Как разбить список на несколько списков с определенными диапазонами?

Question

У меня есть список, который нужно разделить на 4 отдельных списка с максимальным размером 21 в зависимости от количества элементов в исходном списке.

В основном списке может быть от 1 до 84 наименований.

Я хочу, чтобы элементы начинались в a и заполнялись максимум до 21 в a, b, c, d

У меня есть следующий код, который может разделить элементы без проблем, но я хочу знать, есть ли более короткий способ сделать это. Я повторяю код много, кроме диапазона.

codes = [x for x in range(80)] # range anywhere between 1-84

print(len(codes))

a = []
b = []
c = []
d = []

for i in range(0, 21):
    try:
        a.append(codes[i])
    except IndexError:
        pass

for i in range(21, 42):
    try:
        b.append(codes[i])
    except IndexError:
        pass

for i in range(42, 63):
    try:
        c.append(codes[i])
    except IndexError:
        pass

for i in range(63, 84):
    try:
        d.append(codes[i])
    except IndexError:
        pass

print(len(a), len(b), len(c), len(d))
print(a)
print(b)
print(c)
print(d)

До этого у меня был этот код, который отлично работает на все 84 позиции, так как порядок не важен.

    a = []
    b = []
    c = []
    d = []

    for a1, b1, c1, d1 in zip(*[iter(codes)]*4):
        a.append(a1)
        b.append(b1)
        c.append(c1)
        d.append(d1)

Однако, если я скажу 4 пункта, он добавит 1 к каждому

a = [0]
b = [1]
c = [2]
d = [3]

Я бы хотел получить

a = [0, 1, 2, 3]
b = []
c = []
d = []

Answer 1

@ SajalPreetSinghs ответ правильный для простого варианта использования, как у вас!

Но он также имеет некоторые недостатки , когда дело доходит до масштабируемость . Например:

• Когда вам нужно 20 подсписков вместо фактических 4. Вам нужно будет добавить еще 16 строк и указать 2 разных числа в каждой строке!
• Теперь представьте, что вы уже расширили код до 20 подсписков, но теперь вы хотите, чтобы максимальное число элементов в подсписке было 37 вместо 21 - вам нужно изменить 2 * 20 = 40 номеров!

Так что, если вы хотите что-то с лучшей масштабируемостью, вы можете использовать следующий код, который использует генераторы :

import random

# CODE #
def sublist(master, list_of_subs, max_items_per_list):
    def sublist_generator():
        for sublist in list_of_subs:
            yield sublist

    sublist = sublist_generator()
    current = next(sublist)
    for code in codes:
        current.append(code)

        if len(current) == max_items_per_list: # current list is full
            current = next(sublist) # so let current point to the next list

# SETUP AND USAGE #
start = 1
stop = random.randint(2, 85) # generate random int inclusively 2 and 85
codes = [x for x in range(start, stop)] # stop is exclusive: range(1, 85) == [1, 2, ..., 84]

a, b, c, d = [], [], [], []
sublists = [a, b, c, d] # *1

sublist(codes, sublists, 21)
for sublist in sublists:
    print(sublist)

Лучшая масштабируемость:

• Если вы хотите изменить количество элементов в подсписке, вам нужно только передать новое максимальное число.

• Если вы хотите увеличить количество подсписков, вам нужно только добавить их в переменную sublists, которую вы передаете функции (см. *1).

• Если вам нужен этот код чаще, это не проблема, потому что вы можете удобно вызывать функцию.

Надеюсь, это кому-нибудь поможет!

Приветствия
winklerrr

Answer 2

Вы можете просто использовать подсписок

a = codes[0: 21]
b = codes[21:42]
c = codes[42:63]
d = codes[63:84]

Это будет хорошо для вашего требования

Answer 3

Что бы я сделал, просто поместил бы это в один цикл.

, если индекс от 0 до 83

for i in range(0,84):

   if i>=0 and i<=20:
      a.append(codes[i])
   elif i>20 and i<=41:
      b.append(codes[i])
   elif i>41 and i<=62:
      c.append(codes[i])
   elif i>62 and i<=83:
      d.append(codes[i])

Answer 4

Вы можете использовать функцию zip_longest из itertools

from itertools import zip_longest

master_list = range(0, 84)

iterables = [iter(master_list)] * 21
slices = zip_longest(*iterables, fillvalue=None)

for slice in slices:
    print("slice", slice)

# slice (0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20)
# slice (21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41)
# slice (42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62)
# slice (63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83)

Answer 5

Вы можете сделать это с помощью понимания списка:

MAX_SIZE = 21

l = list(range(80))

l1,l2,l3,l4 = [l[i*MAX_SIZE: (i+1)*MAX_SIZE] for i in range(4)] 


#l1=[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19,20],
# .... 
#l4=[63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79]