检查字符串是否只包含某些字母

Question

在python中如何创建一个函数来检查用户输入的字符串是否只包含字母a和b？  字符串可以与用户想要的一样长，但它应该只包含字母a和b的组合。 感谢。

Answer 1

使用正则表达式(a|b)*$检查字符串是否仅包含a和b

CODE：

import re
import sys

prog = re.compile('(a|b)*$')

str ="""aaabbb"""

if prog.match(str):
    print "String matches"
else:
    print "String doesn't match"

Answer 2

除了使用正则表达式之外，您还可以使用generator expression检查all字符是in字符串"ab"。

>>> all(c in "ab" for c in "foobar")
False
>>> all(c in "ab" for c in "aaabaababbba")
True