Zero Matrix

gistfile1.txt

def zero_matrix(matrix):
  """Take a matrix m x n, if any val is zero set the entire row/column to zero"""
  zero_loc = []
  
  for j in range(len(matrix)):
    for i in range(len(matrix[0])):
      if matrix[j][i] == 0:
        zero_loc.push((j,i))
        
  for j, i in zero_loc:
    matrix[j] = [0 for val in range(len(matrix[j])]
    for row in matrix:
      matrix[row][i] = 0
  
  return matrix