题解 | 矩阵和向量的点积

import numpy as np

def matrix_vector_dot_product(matrix, vector):
    matrix_arr = np.array(matrix)
    vector_arr = np.array(vector)
    if matrix_arr.shape[1] != vector_arr.shape[0]:
        return -1
    result = np.dot(matrix_arr, vector_arr)
    return result.tolist()


# 主程序
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)

    # 输出结果
    print(output)