协助numpy savetxt功能

时间:2014-05-24 05:30:29

标签: python numpy

我是python的新手,我正在努力弄清楚如何将数据输出到文件中。

我的脚本的以下部分效果很好,但我希望能够将打印数据输出到文本文件。

x = np.linspace(141, 144.5, 500)
print x
y = np.linspace(-38.53, -38.53, 500)
z = np.linspace(0, 0, 500)

gz = tesseroid.gz(x,y,z,model)
print gz

作为对此的补充;当它在我的终端打印时,它打印数据如下......

(x)的

1 2 3 4
5 ... 500

(GZ)

1 2 3 4
5 ... 500

...但是我希望它能在一个列中输出数据,如下所示:

1
2
3
4
5
...
8

如果它们都可以在同一个文件中会更好......

1 1
2 2
3 3
4 4
5 5
...
500 500

...但是这不是必需的,因为我可以轻松地从那里手动操作它们。

非常感谢,如果这是一个非常简单的问题,我会道歉;这是我用我有限的经验无法弄清楚的事情。

编辑:请注意,它不一定要使用numpy savetext函数。如果有更简单的方法来执行此任务,那么我非常愿意使用它。

1 个答案:

答案 0 :(得分:1)

正如标题所示,您可以使用np.savetxt()保存在一列中:

np.savetxt('x.txt', x.ravel())
np.savetxt('gz.txt', gz.ravel())

或保存在两列中:

np.savetxt('x_gz.txt', np.vstack((x.ravel(), gz.ravel())).T)