thiagowfx's avatar

Β¬ just serendipity πŸ€ (not just serendipity)

LeetCode #2788: Split Strings by Separator

β€’ 66 words β€’ 1 min

LeetCode #2788: Split Strings by Separator:

Flatten with itertools:

python
import itertools

class Solution:
    def splitWordsBySeparator(self, words: List[str], separator: str) -> List[str]:
        return [el for el in list(itertools.chain.from_iterable(word.split(separator) for word in words)) if len(el) > 0]

Manual:

python
import itertools

class Solution:
    def splitWordsBySeparator(self, words: List[str], separator: str) -> List[str]:
        ans = []
        for word in words:
            for token in word.split(separator):
                if token:
                    ans.append(token)
        return ans