我正在编写一个函数,它将返回一个平方数列表,但如果该函数接受参数('apple')或(range(10))或一个列表,则将返回一个空列表。我已经完成了第一部分,但不知道如何返回空集,如果参数n不是整数-我一直得到一个错误: unorderable type: str() > int()我知道字符串不能与整数进行比较,但我需要它来返回空列表。
def square(n):
return n**2
def Squares(n):
if n>0:
mapResult=map(square,range(1,n+1))
squareList=(list(mapResult))
else:
squareList=[]
return squareList
发布于 2013-07-06 11:13:48
您可以使用python中的type
函数来检查变量的数据类型。为此,您可以使用type(n) is int
来检查n
是否是您想要的数据类型。而且,map
已经返回了一个列表,所以不需要强制转换。因此..。
def Squares(n):
squareList = []
if type(n) is int and n > 0:
squareList = map(square, range(1, n+1))
return squareList
发布于 2013-07-06 11:09:13
你可以使用or
将所有导致返回空列表的条件链接到一个条件中。也就是说,如果它是一个列表,或者等于'apple'
,或者等于range(10)
或n < 0
,那么返回一个空列表。否则,返回mapResult。
def square(n):
return n**2
def squares(n):
if isinstance(n,list) or n == 'apple' or n == range(10) or n < 0:
return []
else:
return list(map(square,range(1,n+1)))
isinstance
检查n
是否是list
的实例。
一些测试用例:
print squares([1,2])
print squares('apple')
print squares(range(10))
print squares(0)
print squares(5)
获取
[]
[]
[]
[]
[1, 4, 9, 16, 25]
>>>
https://stackoverflow.com/questions/17499268
复制相似问题