RosettaCodeData/Task/Identity-matrix/Python/identity-matrix-2.py
Ingy döt Net 776bba907c Sync
2013-10-27 22:24:23 +00:00

11 lines
277 B
Python

>>> def identity(size):
... return {(x, y):int(x == y) for x in range(size) for y in range(size)}
...
>>> size = 4
>>> matrix = identity(size)
>>> print('\n'.join(' '.join(str(matrix[(x, y)]) for x in range(size)) for y in range(size)))
1 0 0 0
0 1 0 0
0 0 1 0
0 0 0 1
>>>