From 4a3514afd078df234f7acc68eed0b893fc639ccb Mon Sep 17 00:00:00 2001 From: Steven Date: Mon, 27 May 2024 12:01:17 +0100 Subject: add solution week 271 task 1 in python --- challenge-271/steven-wilson/python/ch-1.py | 23 +++++++++++++++++++++++ 1 file changed, 23 insertions(+) create mode 100644 challenge-271/steven-wilson/python/ch-1.py diff --git a/challenge-271/steven-wilson/python/ch-1.py b/challenge-271/steven-wilson/python/ch-1.py new file mode 100644 index 0000000000..b746c4b739 --- /dev/null +++ b/challenge-271/steven-wilson/python/ch-1.py @@ -0,0 +1,23 @@ +#!/usr/bin/env python3 + +from operator import itemgetter + + +def maximum_ones(matrix): + ''' Given a m x n binary matrix, return the row number containing maximum + ones, in case of more than one rows then return smallest row number. + >>> maximum_ones([ [0, 1],[1, 0],]) + 1 + >>> maximum_ones([ [0, 0, 0],[1, 0, 1],]) + 2 + >>> maximum_ones([ [0, 0],[1, 1],[0, 0],]) + 2 + ''' + row_sum = ((n, sum(row)) for n, row in enumerate(matrix, start=1)) + return sorted(row_sum, key=itemgetter(1), reverse=True)[0][0] + + +if __name__ == "__main__": + import doctest + + doctest.testmod(verbose=True) -- cgit