平方列表中的所有元素

Question

有人告诉我

编写一个函数square（a），该函数接受一个数字数组a并返回一个包含每个平方值的数组。

起初，我有

def square(a):
    for i in a: print i**2

但是，由于我正在打印，而且没有像被问到的那样返回，因此这不起作用。所以我尝试了

    def square(a):
        for i in a: return i**2

但这仅平方我数组的最后一个数字。我如何才能使整个列表平方？

Answer 1

您可以使用列表理解：

def square(list):
    return [i ** 2 for i in list]

或者您可以map：

def square(list):
    return map(lambda x: x ** 2, list)

或者，您可以使用发电机。它不会返回列表，但是您仍然可以迭代它，并且由于不必分配整个新列表，因此它可能比其他选项更节省空间：

def square(list):
    for i in list:
        yield i ** 2

或者，您也可以执行无聊的旧for循环，尽管这并不像某些Python程序员所希望的那样惯用：

def square(list):
    ret = []
    for i in list:
        ret.append(i ** 2)
    return ret