机器学习每日一题000-矩阵和向量的乘法python实现

题目内容

将矩阵和向量的点积计算过程用 Python 函数实现。如果操作有效，该函数应返回一个代表结果向量的列表；如果矩阵和向量的维度不兼容，则返回 -1。矩阵（一个列表的列表）只有在其列数等于向量（一个列表）的长度时，才能与该向量进行点积运算。例如，一个 $n\times m$ 的矩阵需要一个长度为 $m$ 的向量。

例子

Input:
a = [[1, 2], [2, 4]], b = [1, 2]
Output:
[5, 10]
Reasoning:
Row 1: (1 * 1) + (2 * 2) = 1 + 4 = 5; Row 2: (1 * 2) + (2 * 4) = 2 + 8 = 10

解答代码

Python List 实现

def matrix_dot_vector(a: list[list[int|float]], b: list[int|float]) -> list[int|float]:result = []if len(a[0]) != len(b):return -1for row in a:dot = 0for i in range(len(b)):dot += row[i] * b[i]result.append(dot)return resultif __name__ == "__main__":a = [[1, 2], [3, 4]]b = [1, 2]print(matrix_dot_vector(a, b))# a = (input().strip().split())