equivalent of j in numpy,1j * np.arange(5)