我该如何使用列表理解来重写此功能?结果函数应该只有两行:定义行和包含列表理解表达式的返回行。
def processList(listOfNumbers):
result = []
for i in listOfNumbers:
if i<0:
result.append(i*i)
else:
result.append((i*i)+1)
return result
答案 0 :(得分:1)
您可能正在尝试这样做:
def processList(listOfNumbers):
return [i ** 2 if i < 0 else i ** 2 + 1 for i in listOfNumbers]