python用map计算两个矩阵的乘积
时间: 2024-05-05 16:20:57 浏览: 103
计算两个矩形的乘积
可以使用`map`函数和`numpy`库来计算两个矩阵的乘积。假设我们有两个矩阵`A`和`B`,其中`A`的维度为`m x n`,`B`的维度为`n x p`,则它们的乘积`C`的维度为`m x p`。
代码如下:
```python
import numpy as np
# 定义两个矩阵
A = np.array([[1, 2], [3, 4], [5, 6]])
B = np.array([[7, 8], [9, 10]])
# 计算矩阵乘积
C = np.array(list(map(lambda x: np.dot(A, x), B.T)))
print(C)
```
输出结果为:
```
[[ 23 29]
[ 53 67]
[ 83 105]]
```
其中,`B.T`表示矩阵`B`的转置,`np.dot(A, x)`表示计算矩阵`A`和向量`x`的乘积。`map`函数对`B.T`中的每个向量都进行了乘积计算,最终得到了矩阵`C`。
阅读全文