使用python驗證兩個字符的代碼

Question

我想驗證一個字符串是否是有效的代碼;特別是它應該是一對兩個字符。想想一個2個字符的國家代碼。使用re模塊，我想出了以下情況：

valid = re.compile('([a-zA-Z]){2,2}') 
if valid.match(s) and len(s) == 2: 
    return True 
else: 
    return False 

我覺得這是不是在所有優化。我怎樣才能優化這個測試？

Answer 1

只返回一個簡單的測試結果爲str.isalpha()加上長度測試：

return len(s) == 2 and s.isalpha() 

在默認C區域字節字符串（例如，你沒有與locale模塊更改地區）是」只有對於包含兩個ASCII字母（大寫或小寫）的字符串才適用。

如果必須使用正則表達式，然後錨表達：

return re.match('^[a-zA-Z]{2}$', s) is not None 

的is not None測試也給你一個布爾值。 ^錨定暗示re.match()，但明確表示不會造成傷害。

不使用正則表達式快但是：

>>> import timeit, re, random 
>>> tests = [''.join([chr(random.randrange(256)) for _ in range(2)]) for _ in range(1000)] 
>>> def test_str_isalpha(s): 
...  return len(s) == 2 and s.isalpha() 
... 
>>> def test_regex(s, pattern=re.compile('^[a-zA-Z]{2}$')): 
...  return pattern.match(s) is not None 
... 
>>> timeit.timeit('for s in tests: test(s)', 'from __main__ import tests, test_str_isalpha as test', number=10000) 
2.140676975250244 
>>> timeit.timeit('for s in tests: test(s)', 'from __main__ import tests, test_regex as test', number=10000) 
3.8515820503234863