python矩阵操作 - 对每个元素进行平方

Question

我如何采用如下矩阵:

matrix = [[2,0,2],[0,2,0],[2,0,2]]并将其转换为[[4,0,4],[0,4,0],[4] ,0,4]

列表中的所有元素都变为平方？我正在尝试列表理解.

Answer 1

使用列表理解:

>>> lis = [[2,0,2],[0,2,0],[2,0,2]]
>>> [[y**2 for y in x] for x in lis]
[[4, 0, 4], [0, 4, 0], [4, 0, 4]]