Writing and reading complex numbers using numpy.savetxt and numpy.loadtxt

后端 未结 1 482
野性不改
野性不改 2021-01-20 08:51

I need to write and read complex numbers. I would like to use numpy.savetxt and numpy.loadtxt to do so. Since the code that I have written is rathe

1条回答
  •  小蘑菇
    小蘑菇 (楼主)
    2021-01-20 09:11

    Thanks Warren Weckesser! The link you suggested helped me a lot. I now have two working scripts: one for writing complex numbers using numpy.savetxt and one for reading/loading the complex numbers from the file using numpy.loadtxt.

    For future references, the codes are listed below.

    Writing:

    import numpy
    
    d1 = -0.240921619563-0.0303165074169j
    d2 = -0.340921619563-0.0403165074169j
    d3 = -0.440921619563-0.0503165074169j
    d4 = -0.540921619563-0.0603165074169j
    
    array = numpy.array([d1, d2, d3, d4])
    
    save = open("test.dat","w")
    numpy.savetxt(save, array.reshape(1, array.shape[0]), newline = "\r\n", fmt = '%.4f%+.4fj '*4)
    
    save.close()
    

    Reading/Loading:

    import numpy
    
    coeffs = numpy.loadtxt("test.dat", dtype = numpy.complex128)
    

    0 讨论(0)
提交回复
热议问题