Julia 如何把复数向量保存成 python 中的 .npy 文件？ - FAQ

可以利用 Julia 的 PyCall 包和 Python 的 numpy 库，将 Julia 的复数向量转换为 Python 的 numpy 数组，然后使用 numpy.save 函数保存保存为 .npy 文件。

Julia 的示例代码如下：

using PyCall
np = pyimport("numpy")
# 创建复数向量
temp = ComplexF64[1+2im, 3+4im]
# 方法一：将复数向量转换为numpy array
# z_numpy =np.array(temp)
# 保存为.npy文件
# np.save("complex.npy",z_numpy)

# 方法二：将复数分解为实部和虚部
real_part = real(temp)
imaginary_part = imag(temp)
# 创建一个二维数组来保存实部和虚部
z_numpy = np.array([real_part, imaginary_part])
# 保存为 .npy 文件
np.save("complex.npy", z_numpy)

使用 python 库 numpy 读取 npy 文件，示例代码如下：

import numpy as np
# 方法一读取： 加载 .npy 文件
# data = np.load("complex.npy")
# 打印结果
# print(data)

# 方法二读取： 加载 .npy 文件
data = np.load("complex.npy")
# 提取实部和虚部
real_part = data[0]
imaginary_part = data[1]
# 组合成复数向量
complex_vector = real_part + 1j * imaginary_part
print(complex_vector)

全部回答