Python - 將 for 循環變成單行循環

Question

我正在嘗試創建一個矩陣乘法標量函數，沒有任何庫。 它必須包括列表理解：

A = [[1,2],[3,4]] # 2by2 matrix

scalar = 2 #positive int

product = []

for row in A:
    
    temp = []
    
    for element in row:
        temp.append(scalar * element)
    
    product.append(temp)

print(product)

Answer 1

這是一個可能的解決方案：

A = [[1,2],[3,4]] # 2by2 matrix

scalar = 2 #positive int

product = [[i*scalar for i in sublist] for sublist in A]

print(product)

Answer 2

或者，您可以使用 lambdas 和map獲得一些樂趣：

product = [*map(lambda x: [*map(lambda y: y * scalar, x)], A)]

在線試試吧！