【数据分析面试】32.矩阵元素求和 (Python： for…in…语句)

题目：矩阵元素求和 （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 ... 结构的语法使得在处理循环迭代时非常简洁和易读。

更多详细答案可关注公众号查阅。