Function squaring 2-d array python

Question

I have a function that takes in any 2-d array and return a 2-d array (the same format as the array being implemented) but the values are squared. i.e [[1,2],[3,4]] -----> [[1,4],[9,16]]

my code so far:

m0 = [[1,2],[3,4]]
empty_list = []
for x in m0:
   for i in x:
     empyt_list.append(x**2)

This gives me a 1-d array but how would i return a 2-d array as the imputed value?

Answer 1

You can make a recursive function to handle any depth of nested lists:

def SquareList(L):
    if type(L) is list:
        return [SquareList(x) for x in L]
    else:
        return L**2

Example:

 > print(SquareList([1,[3],[2,[3]],4]))
 [1, [9], [4, [9]], 16]