题解 | 矩阵和向量的点积

def matrix_vector_dot_product(matrix, vector):
    # 补全代码
    import numpy as np 
    matrix = np.array(matrix)
    vector = np.array(vector)
    row, col = matrix.shape
    b = vector.shape[0]   #列
    if col !=b:
        return None
    else: 
        result = np.zeros(row, dtype=int)
        for i in range(row):
            for j in range(col):
                result[i] += matrix[i][j]*vector[j]
        return result


# 主程序
if __name__ == "__main__":
    # 输入矩阵和向量
    matrix_input = input()
    vector_input = input()

    # 处理输入
    import ast
    matrix = ast.literal_eval(matrix_input)
    vector = ast.literal_eval(vector_input)

    # 调用函数计算点积
    output = matrix_vector_dot_product(matrix, vector)
    
    # 输出
    if output is None:
        print("-1")
    else:
        # 输出结果，添加方括号
        print(f"[{', '.join(map(str, output))}]")