今天写作业的时候突然想到,一直使用isdigit()方法来处理用户的输入选择是不是数字,但是如果用户输入的是负数呢,会不会导致bug?
然后我就试了一下,居然不报错。。。然后我就纳闷了,赶紧试了一下:
先来看看str类的.isdigit()方法的文档。
1 def isdigit(self): # real signature unknown; restored from __doc__
2 """
3 S.isdigit() -> bool
4
5 Return True if all characters in S are digits
6 and there is at least one character in S, False otherwise.
7 """
8 return False
很显然'-10'.isdigit()返回False是因为'-'不是一个digit。
然后我就想怎么才能让负数也正确的判断为整数呢,下面是从网上找到的答案,在这里记录下来。
1 num = '-10'
2 if (num.startswith('-') and num[1:] or num).isdigit():
3 print(num是整数)
4 else:
5 print(num不是整数)
正则表达式法:
1 num = '-10'
2 import re
3 if re.match(r'^-?(.d+|d+(.d+)?)', num):
4 print(num是整数)
5 else:
6 print(num不是整数)
更Pythonic的方法:
1 num = '-10'
2 if num.lstrip('-').isdigit():
3 print(num是整数)
4 else:
5 print(num不是整数)