Python列表理解转换

时间:2019-03-06 19:22:32

标签: python list-comprehension

我该如何使用列表理解来重写此功能?结果函数应该只有两行:定义行和包含列表理解表达式的返回行。

def processList(listOfNumbers):
    result = []
    for i in listOfNumbers:
        if i<0:
            result.append(i*i)
        else:
            result.append((i*i)+1)
    return result

1 个答案:

答案 0 :(得分:1)

您可能正在尝试这样做:

def processList(listOfNumbers):
    return [i ** 2 if i < 0 else i ** 2 + 1 for i in listOfNumbers]