In [8]:
import numpy as np
import mpi4py.MPI
import os # for IO-paths
import matplotlib.pyplot as plt
plt.style.use('bmh')

In [2]:
%matplotlib inline

In [5]:
for i in range(4):
    print(i)


0
1
2
3

In [9]:
a = np.arange(4)

In [11]:
print(a%2)


[0 1 0 1]