You could achieve this through re.sub function.

import re
def stripComments(code):
    code = str(code)
    return re.sub(r'(?m)^ *#.*n?', '', code)

print(stripComments("""#foo bar
bar foo
# buz"""))

(?m) enables the multiline mode. ^ asserts that we are at the start. <space>*# matches the character # at the start with or without preceding spaces. .* matches all the following characters except line breaks. Replacing those matched characters with empty string will give you the string with comment lines deleted.

For my future reference.

def remove_comments(lines: list[str]) -> list[str]:
    new_lines = []
    for line in lines: 
        if line.startswith("#"):  # Deal with comment as the first character
            continue

        line = line.split(" #")[0]
        if line.strip() != "":
            new_lines.append(line)

    return new_lines


print(remove_comments("Hello #World!nnI have a question # that #".split('n')))
>>> ['Hello', 'I have a question']

This implementation has benefit of not requiring the re module and being easy to understand. It also removes pre-existing blank lines, which is useful for my use case.

def remove_comments(filename1, filename2):
    """ Remove all comments beginning with # from filename1 and writes
    the result to filename2
    """

    with open(filename1, 'r') as f:
        lines = f.readlines()

    with open(filename2, 'w') as f:
        for line in lines:
            # Keep the Shebang line
            if line[0:2] == "#!":
                f.writelines(line)
            # Also keep existing empty lines
            elif not line.strip():
                f.writelines(line)
            # But remove comments from other lines
            else:
                line = line.split('#')
                stripped_string = line[0].rstrip()
                # Write the line only if the comment was after the code.
                # Discard lines that only contain comments.
                if stripped_string:
                    f.writelines(stripped_string)
                    f.writelines('n')

def removeComments(string: str):
    """Takes something like:

    import math

    math.pow(2,3) # 2 to the power of 3

    and outputs:

    import math
    
    math.pow(2,3) 

    Maintains trailing whitespace
    """
    isInComment=False
    returnVal = ""
    for char in string:
        if char == '#':
            isInComment=True
        elif char == "n":
            isInComment=False
        if not isInComment:
            returnVal+=char
    return returnVal