写一个可以进行矩阵相乘的python代码
时间: 2024-05-02 08:21:41 浏览: 87
下面是一个简单的 Python 代码,用于矩阵相乘:
```python
def matrix_multiply(matrix_1, matrix_2):
# 获取矩阵 1 的行数和列数
rows_1 = len(matrix_1)
cols_1 = len(matrix_1[0])
# 获取矩阵 2 的行数和列数
rows_2 = len(matrix_2)
cols_2 = len(matrix_2[0])
# 如果矩阵 1 的列数不等于矩阵 2 的行数,则无法相乘
if cols_1 != rows_2:
return None
# 创建一个与结果矩阵相同大小的零矩阵
result = [[0 for col in range(cols_2)] for row in range(rows_1)]
# 进行矩阵相乘
for i in range(rows_1):
for j in range(cols_2):
for k in range(cols_1):
result[i][j] += matrix_1[i][k] * matrix_2[k][j]
return result
# 示例
matrix_1 = [[1, 2], [3, 4]]
matrix_2 = [[5, 6], [7, 8]]
result = matrix_multiply(matrix_1, matrix_2)
print(result) # 输出 [[19, 22], [43, 50]]
```
该代码中,我们首先获取了两个矩阵的行数和列数,然后判断它们是否可以相乘。如果可以相乘,我们就创建一个与结果矩阵相同大小的零矩阵,并使用三重循环进行矩阵相乘。最后返回结果矩阵。
阅读全文