[4 kyu] Strip Comments

[4 kyu] Strip Comments

Question

Sample Tests

My Answer (accepted)

def solution(string,markers):
    string_piece = string.split("\n")
    return_piece = list()
    for unit in string_piece:
        unit_split = list(unit) 
        for (index, c) in enumerate(unit_split):
            if c in markers:
                unit_split = unit_split[:index]
                break
        tmp = "".join(unit_split)
        return_piece.append(tmp.strip())
    print(return_piece)
    return "\n".join(return_piece)

Suggested Answer

def solution(string,markers):
    parts = string.split('\n')
    for s in markers:
        parts = [v.split(s)[0].rstrip() for v in parts]
    return '\n'.join(parts)