python 核心编程第六章习题16 矩阵加法和乘法

最新推荐文章于 2024-06-30 17:19:02 发布

原创最新推荐文章于 2024-06-30 17:19:02 发布 · 3.5k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #编程 #list

python 核心编程专栏收录该内容

11 篇文章

订阅专栏

本文提供了一种使用Python进行矩阵加法及乘法运算的方法。通过定义两个函数：mar_add用于矩阵相加，mar_multiply用于矩阵相乘，实现了基本的矩阵运算逻辑。示例中展示了如何对两个不同维度的矩阵进行乘法运算。

很简单，但是写的有点复杂感觉，有没有朋友有简单一些的方法，还请不吝赐教～

#!/usr/bin/python 

def mar_add(list1,list2):
	m = len(list1)
	n = len(list1[0])
	if m!= len(list2):
		print "error ! two list must have same dimson"
		return 
	if n!= len(list2[0]):
		print "error ! two list must have same dimson"
		return 
	new_list = []

	for i in range(m):
		new_list.append([])
		for j in range(n):
			new_list[i].append(list1[i][j] + list2[i][j])
	
	return new_list

def mar_multiply(list1,list2):
	m = len(list1)
	k = len(list1[0])
	n = len(list2[0])

	if k != len(list2):
		print "error ! dimension error !"
		return 
	new_list = []
	for i in range(m):
		new_list.append([])
		for j in range(n):
			new_list[i].append(0)
			for x in range(k):
				new_list[i][j] += list1[i][x] * list2[x][j]
	
	return new_list



if __name__ == "__main__":
	list1 = [[1,2],[0,1],[2,3],[5,6]]
	list2 = [[0,1,2],[1,1,0]]

	print mar_multiply(list1,list2)