Python map 函数查找 1 数量最多的行

pythonserver side programmingprogramming更新于 2025/10/24 0:12:17

在本教程中，我们将编写一个程序，使用 map 函数从矩阵中查找 1 数量最多的行。

假设我们有以下矩阵。

matrix = [ [0, 0, 1], [1, 1, 1], [1, 1, 0] ]

我们可以用不同的方式编写程序。但是，使用 map 函数，我们将遵循以下步骤。

初始化矩阵。
使用 map 函数查找每行中 1 的数量。将它们存储在列表中。
从列表中打印最大值。

示例

## 初始化矩阵
matrix = [
   [0, 0, 1],
   [1, 1, 1],
   [1, 1, 0]
]
## 函数用于查找一行中 1 的数量
def number_of_ones(row):
   count = 0
   for i in row:
      if i is 1:
         count += 1
   return count
## 查找每行中 1 的数量
## map 返回一个我们转换为列表的对象
ones_count = list(map(number_of_ones, matrix))
## 打印列表中最大数字的索引
print(ones_count.index(max(ones_count)))