将FFT脚本数据导出到文本文件中

Question

我对老师在学生实验课上给我们的剧本有疑问。该脚本用Python 2编写，从mic.txt文件导入数据，执行FFT变换，然后绘制数据。我想要的是，脚本还将生成一个包含两列的txt文件，其中包含打印的数据编号，这样我就可以使用结果进行进一步分析

谢谢你的帮助，马修

from scipy.fftpack import fft,ifft
import matplotlib.pyplot as plt
from scipy.signal import blackman

data =  np.loadtxt("mic.txt")
x = data[:,0]
y = data[:,1]

fy = fft(y)
ft = np.fft.fft(y)

print np.sum(y),"==",fy[0]

n = len(x)
t = x[-1]
fx = np.linspace(0,n/t,n)

plt.plot(fx[0:n/2],np.abs(fy[0:n/2]))

plt.xlabel("frequency (Hz)")
plt.show()

Answer 1

如果两列长度相等，则可以按如下方式将其写入文本文件：

with open('text_file.txt','w') as f:
    for a,b in zip(variable_1, variable_2):
        f.write(str(a) + ',' + str(b) + '\n')
f.close()