Equivalent of j in Numpy,1j * np.arange(5)