Python

Python Tips and Tricks

Array enumeration

For numpy

a = np.array([[1, 2], [3, 4]])
for index, x in np.ndenumerate(a):
    print(index, x)

For usual python

a = np.array([[1, 2], [3, 4]])
for index, x in enumerate(a):
    print(index, x)

Context

 with open(filename, "w") as f:
        f.write("hello!\n")
    # close automatic, even if exception