python - Python将txt文件保存在不同的文件夹中

Question

当我从桌面运行脚本时，以下脚本运行良好并将 txt 输出保存在桌面中。但是，我想将 txt 文件保存到我的 Documents 中名为 ASCII 的新文件夹中。我怎样才能给出这样做的命令。8phases.txt 有以下几行 -

-1  1   -1

-1  1   1

1   1   1

1   -1  1

-1  -1  -1

1   1   -1

1   -1  -1

-1  -1  1

剧本-

import numpy as np
import matplotlib.pyplot as plt

D=12
n=np.arange(1,4)
x = np.linspace(-D/2,D/2, 3000)
I = np.array([125,300,75])
phase = np.genfromtxt('8phases.txt')

I_phase = I*phase

for count,i in enumerate(I_phase):
    F = sum(m*np.cos(2*np.pi*l*x/D) for m,l in zip(i,n))
    s = np.column_stack([x,F])
    np.savetxt((str(count)+'.txt'),s)

请提供任何帮助-

Answer 1

您可能应该在 savetxt 方法的参数中提供完整路径，例如：

np.savetxt(r"C:\ASCII\%s.txt" % count,s)

Answer 2

您可以尝试以下方法

from os.path import osp
userdoc = osp.join(osp.expanduser("~"),'Documents')
np.savetxt(osp.join(userdoc, "%s.txt" % count),s)