Разделить строку на сегменты с помощью Python

Question

Я пытаюсь придумать код, который может взять строку и разбить ее на сообщения с количеством символов n. Каждое сообщение не может превышать n, но может содержать не более n символов, поскольку слова не следует разбивать. Например, «Это пример сообщения» и n = 10. Этот код вернется 'Это' 'пример' 'Сообщение'

Любые предложения о том, как я могу подойти к этому?

import math 

def solution(S, K):
    x = math.ceil(len(S)/K)
    y = S.split()

    lists = [[] for i in range(x)]

    for i in lists:
        while len(i) <= K:
            i.append(y[b])
            b+=1

x - это количество сообщений, которые я считаю необходимыми. Может кто-нибудь объяснить мне, как я могу завершить мой код?

Answer 1

Для этого предназначена библиотека textwrap .

>>> from textwrap import wrap
>>> print(wrap('This is an example message', 10))
['This is an', 'example', 'message']