题解 | 标量的矩阵乘法

from typing import List, Union
import numpy as np

def scalar_multiply(matrix: List[List[Union[int, float]]], scalar: Union[int, float]) -> List[List[Union[int, float]]]:
    arr = np.array(matrix)
    arr2 = arr * scalar
    return arr2.tolist()
    

def main():
    matrix = eval(input())
    scalar = float(input())
    result = scalar_multiply(matrix, scalar)
    print(result)

if __name__ == "__main__":
    main()