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

Question

[BlackAlexYT]

Дается массив цисел mas, в котором находятся нули, нужно разбить массив на подмассивы, используя "0" как разделитель. Например mas=[1,2,3,0,5,3,6,0,7,5] должен стать [[1,2,3],[5,3,6],[7,5]]

Answer 1

[LXSTVAYNE]

Решение если у тебя бесконечные ресурсы:

mas = [1, 2, 3, 0, 5, 3, 6, 0, 7, 5]
res = [list(map(int, list(el))) for el in ''.join(map(str, mas)).split('0')]
print(res)

Одно из решений питониста джуна:

mas = [1, 2, 3, 0, 5, 3, 6, 0, 7, 5]

res = []

temp = []
for el in mas:
    if el != 0:
        temp.append(el)
    else:
        res.append(temp)
        temp = []
else:
    res.append(temp)

print(res)

И, наконец, решение питониста повидавшего жизнь. Используется библиотека more_itertools:

from more_itertools import split_at


mas = [1, 2, 3, 0, 5, 3, 6, 0, 7, 5]

res = list(split_at(mas, lambda x: not x))

print(res)

Если есть ещё какие-нибудь интересные решения, можете приложить в комментарии к ответу.

Answer 2

[Сергей П]

Без сторонних либ я бы делал так:

def split0(arr: list) -> list:
    res = [[]]
    for x in arr:
        res[-1].append(x) if x else res.append([])
    return res

А вот для любителей функциональщины:

from itertools import takewhile  
def split0(a): 
    it = iter(a) 
    return [[x for x in takewhile(bool, it)] for _ in range(a.count(0) + 1)]

Comments

[LXSTVAYNE]

В последнем решении есть небольшая опечатка, вот исправленный вариант:

from itertools import takewhile  
def split0(a): 
    it = iter(a) 
    return [[x for x in takewhile(bool, it)] for _ in range(a.count(0) + 1)]

[Сергей П]

LXSTVAYNE, ну конечно, спасибо, коллега!
Потерял единичку.

Answer 3

[Vindicar]

Как выглядит твоя попытка решить это?

спойлер

Метод index() у списка в помощь

Comments

[BlackAlexYT]

mas=list(map(int, input().split()))
ind=[-1]
for i in range(len(mas)):
	if mas[i]==0:
		ind.append(i)
ind.append(len(mas))
mas1=[]
for i in range(len(ind)-1):
	mas1.append(mas[ind[i]+1:ind[i+1]])
print(mas1)

Не лучший код, вроде работает, но есть ли вариант решения этой проблемы более "красивый*?

[Vindicar]

BlackAlexYT, вполне себе решение. Хотя можно и так:

mas=list(map(int, input().split()))
parts = []
start, end = 0, 0
while start < len(mas):
    while mas[end] != 0 and end < len(mas):
        end += 1
    parts.append(mas[start:end])
    end += 1 # пропускаем 0
    start = end
print(parts)

Или вот так, но многим не нравится использовать исключения для реализации логики:

mas=list(map(int, input().split()))
parts = []
start= 0
try:
    while start < len(mas):
        end = mas[start:].index(0) + start
        parts.append(mas[start:end])
        start = end + 1
except ValueError: #index() не нашёл 0
    parts.append(mas[start:])
print(parts)