Row With Maximum Ones — Python

# Time:  O(m * n)
# Space: O(1)

# array
class Solution(object):
    def rowAndMaximumOnes(self, mat):
        """
        :type mat: List[List[int]]
        :rtype: List[int]
        """
        return max(([i, mat[i].count(1)] for i in xrange(len(mat))), key=lambda x: x[1])