Python 代码实现矩阵乘法:用户输入矩阵并计算
当然可以!以下是一个可以接受用户输入矩阵并进行计算的代码示例:
import numpy as np
def matrix_multiply(matrix1, matrix2):
result = np.matmul(matrix1, matrix2)
return result
# 获取用户输入的矩阵维度
rows1 = int(input('请输入第一个矩阵的行数: '))
cols1 = int(input('请输入第一个矩阵的列数: '))
rows2 = int(input('请输入第二个矩阵的行数: '))
cols2 = int(input('请输入第二个矩阵的列数: '))
# 创建用户输入的矩阵
matrix1 = np.zeros((rows1, cols1))
matrix2 = np.zeros((rows2, cols2))
print('请输入第一个矩阵的元素:')
for i in range(rows1):
for j in range(cols1):
matrix1[i, j] = int(input(f'请输入第一个矩阵的元素 [{i+1}, {j+1}]: '))
print('请输入第二个矩阵的元素:')
for i in range(rows2):
for j in range(cols2):
matrix2[i, j] = int(input(f'请输入第二个矩阵的元素 [{i+1}, {j+1}]: '))
# 调用函数进行矩阵相乘
result_matrix = matrix_multiply(matrix1, matrix2)
print('结果矩阵:')
print(result_matrix)
这段代码会首先要求用户输入两个矩阵的维度,然后逐个输入矩阵的元素。最后,它将调用之前定义的matrix_multiply()函数来计算两个矩阵的乘积,并将结果打印出来。
请注意,这段代码假设用户输入的元素都是整数。如果你需要支持其他数据类型或有其他特定要求,请提供更多细节,我将为您进行相应的修改。
原文地址: https://www.cveoy.top/t/topic/OGm 著作权归作者所有。请勿转载和采集!