Python size by size matrix that allows input and returns list of lists elements

Question

I\'m trying to create an identity matrix that accepts input and returns an identity matrix(list or lists) of size by size, with only 1\'s and 0\'s without using numpy.

So