让我们重复任务语句:“获取此矩阵的每一行和每一列的最小值”。
好的,所以,如果矩阵有n 行,您应该得到n 最小值,每行一个。听起来很有趣,不是吗?因此,代码将如下所示:
result1 = [<something> for row in matrix]
那么,你需要对每一行做什么?对了,求最小值,超级简单:
result1 = [min(row) for row in matrix]
作为result,您将获得n 值的列表,正如预期的那样。
等等,现在我们只找到了每一行的最小值,但没有找到每一列的最小值,所以我们也这样做!
鉴于您使用的是 Python 3.x,您可以做一些非常了不起的事情。例如,您可以轻松地遍历列:
result2 = [min(column) for column in zip(*matrix)] # notice the asterisk!
zip(*matrix) 中的星号使matrix 的每一行成为zip 的单独参数,如下所示:
zip(matrix[0], matrix[1], matrix[2], matrix[3])
这看起来不太可读,并且取决于matrix 中的行数(基本上,您必须对它们进行硬编码),而星号可以让您编写更简洁的代码。
zip 返回元组,ith 元组包含 所有行的 ith 值,因此这些元组实际上是给定矩阵的列。
现在,您可能会发现这段代码有点难看,您可能希望以更简洁的方式编写相同的内容。果然,你可以使用一些函数式编程魔法:
result1 = list(map(min, matrix))
result2 = list(map(min, zip(*matrix)))
这两种方法是完全等价的。