判断字符串是否为一位数字或两位数字

可以使用正则表达式来判断一个字符串是否为一位数字或两位数字：

import re

def is_one_or_two_digit_number(s):
    return bool(re.match(r'^\d{1,2}$', s))

# 测试
print(is_one_or_two_digit_number('3'))      # True
print(is_one_or_two_digit_number('12'))     # True
print(is_one_or_two_digit_number('123'))    # False
print(is_one_or_two_digit_number('0'))      # True
print(is_one_or_two_digit_number('01'))     # True
print(is_one_or_two_digit_number('001'))    # False

该函数使用了正则表达式 `r'^\d{1,2}$'`，其中：

- `^` 表示匹配字符串的开头；
- `\d` 表示匹配数字；
- `{1,2}` 表示匹配1到2个数字；
- `$` 表示匹配字符串的结尾。

所以该正则表达式可以匹配长度为1或2的纯数字字符串。函数返回值为 True 表示匹配成功，为 False 表示匹配失败。