Identity Matrix In Python