Kiểm tra chuỗi có phải identifier hợp lệ trong Python

Cho chuỗi đại diện cho identifier. Kiểm tra tính hợp lệ theo quy tắc Python:

1. Bắt đầu bằng '_' hoặc chữ cái (A-Z, a-z).
2. Không chứa khoảng trắng.
3. Các ký tự tiếp theo chỉ được là chữ cái, số hoặc '_' (không ký tự đặc biệt $, #, %).

Ví dụ: "_hello_56" → True.

Thuật toán

• Nếu ký tự đầu không phải chữ cái và không phải '_' → False.
• Duyệt từ ký tự thứ 2: nếu không phải alphanumeric và không phải '_' → False.
• Trả về True.

Ví dụ code

def solve(s):
    if not s[0].isalpha() and s[0] != '_':
        return False
    for ch in s[1:]:
        if not ch.isalnum() and ch != '_':
            return False
    return True

id = "_hello_56"
print(solve(id))

Đầu vào

"_hello_56"

Đầu ra

True