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)

Multiprocessing

from multiprocessing import Pool

Context

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

Measure execution timing

import time
start = time.perf_counter()
time.sleep(1)
end = time.perf_counter()
print(end - start)