题目:矩阵元素求和 (Python)
假设给定一个整数矩阵。你的任务是编写一个函数,返回矩阵中所有元素的和。
示例 1:
输入:
matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
输出:
matrix_sum(matrix) -> 45
示例 2:
输入:
matrix = [[-1, -2, -3], [-4, -5, -6], [-7, -8, -9]]
输出:
matrix_sum(matrix) -> -45
该函数应能处理正整数和负整数,并将求和结果作为一个整数返回。
提示:
- 从遍历矩阵中的所有元素开始。
- 确保正确处理正整数和负整数。
答案
解题思路
这个问题的关键在于遍历矩阵中的所有元素,并对它们求和。我们可以通过嵌套循环遍历矩阵的行和列,并累加每个元素的值来解决这个问题。
答案代码
def matrix_sum(matrix):
# 初始化总和为0
total_sum = 0
# 遍历矩阵的行
for row in matrix:
# 遍历每行的元素
for element in row:
# 将每个元素的值加到总和中
total_sum += element
# 返回总和
return total_sum
matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
print(matrix_sum(matrix))
matrix = [[-1, -2, -3], [-4, -5, -6], [-7, -8, -9]]
print(matrix_sum(matrix))
- 我们使用了嵌套循环来遍历矩阵的每个元素。
- 对于每个元素,我们将其值加到总和中。
- 最后,我们返回总和作为结果。
for ... in ...
结构
for ... in ...
是Python中的一种迭代结构,用于遍历可迭代对象中的元素。
在这个结构中,for
关键字用于开始一个循环,in
关键字用于指定要迭代的可迭代对象。循环体内的代码将被执行,针对可迭代对象中的每个元素都会执行一次。
例如:
# 遍历列表中的元素
my_list = [1, 2, 3, 4, 5]
for num in my_list:
print(num)
# 遍历列表中的元素和索引
my_list = ['a', 'b', 'c']
for index, value in enumerate(my_list):
print(index, value)
在这个例子中,num
是循环中的变量,它依次存储了列表 my_list
中的每个元素,然后将其打印出来。
for ... in ...
结构的语法使得在处理循环迭代时非常简洁和易读。
更多详细答案可关注公众号查阅。