Pythonのmap関数を使って「1」が最も多い行を見つけるプログラム

0と1だけで構成された2次元配列（行列）が与えられ、すべての行はあらかじめソートされているものとします。この中から「1」の個数が最も多い行を見つけるのが本記事の目的です。ここでは、Pythonの組み込み関数である map() を活用します。

map() 関数は、関数型プログラミングに用いられるPython組み込みツールの中でも最もシンプルなものの一つです。シーケンスやその他のイテラブル（反復可能オブジェクト）に対して、指定した関数を一括して適用することができます。

実行例

入力 :
入力配列 : [[0, 1, 1, 1, 1], [0, 0, 1, 1, 1], [1, 1, 1, 1, 1], [0, 0, 0, 0, 1]]
1が最も多い行のインデックス = 2

この例では、3行目（インデックス2）の [1, 1, 1, 1, 1] に5つの「1」が含まれており、他のどの行よりも多いため、結果は「2」となります。

アルゴリズム

ステップ1 : map関数を使って、行列の各行ごとの合計値（＝1の個数）を求める。
ステップ2 : 各行の1の合計値を要素とするリストが返される。
ステップ3 : リスト内の最大値のインデックスを出力する。

サンプルコード

# 「1」が最も多い行を見つけるPythonプログラム
def maximumofones(n):
    # map()で各行にsum()を適用し、行ごとの1の個数リストを作成
    max1 = list(map(sum, n))
    print("MAXIMUM NUMBER OF 1's ::>", max1.index(max(max1)))

# ドライバープログラム
if __name__ == "__main__":
    n = [[0, 1, 1, 1, 1], [0, 0, 1, 1, 1], [1, 1, 1, 1, 1], [0, 0, 0, 0, 1]]
    maximumofones(n)

出力結果

MAXIMUM NUMBER OF 1's ::> 2

コードのポイント

list(map(sum, n)) の部分では、map() が行列 n の各行（リスト）に対して sum() 関数を順番に適用します。行の要素は0と1のみなので、各行の合計値はそのまま「1の個数」を表します。その後、max() で最大値を取得し、index() メソッドでその位置（行インデックス）を特定しています。

この手法を使えば、明示的なループ処理を書かずに、簡潔かつ読みやすいコードで目的の行を効率的に見つけることができます。