challenge-271/eric-cheung/python/ch-1.py


1
2
3
4
5
6
7
8
9
10

import numpy as np

## arrMatrix = [[0, 1], [1, 0]]  ## Example 1
## arrMatrix = [[0, 0, 0], [1, 0, 1]]  ## Example 2
arrMatrix = [[0, 0], [1, 1], [0, 0]]  ## Example 3

arrNumOne = [arrMatrix[nLoop].count(1) for nLoop in range(len(arrMatrix))]

print (np.argmax(arrNumOne) + 1)