不同形状数组Python

时间:2017-06-30 19:19:43

标签: python arrays numpy

我正在尝试输出索引及其值。我能够在终端内打印它,但我想将其输入到.txt文件中。这是我目前的代码。我认为提及x& y的形状(528,)是有帮助的,其余的是形状(528,528)。我当前的输出也在代码段之后。

import pencil as pc
import numpy as np

ff = pc.read_var(trimall=True)

x = ff.x
y = ff.y
rho = ff.rho
rhop = ff.rhop
ux = ff.ux
uy = ff.uy

for i in range(528):
    for j in range(528):
        print i,j,rho[i,j],rhop[i,j],ux[i,j],uy[i,j]

我看到的输出是这个,这正是我想要的,但写入.txt文件。

527 524 2.5 3.0999421 1.0025754 3.14249721489 0.00104948277254 0.633309939177
527 525 2.5 3.1118420 1.0046337 0.0 0.000712933516338 0.632900900838
527 526 2.5 3.1237420 1.0032471 0.0 0.000648636250453 0.632665342501
527 527 2.5 3.1356420 1.0008004 0.0 0.000442528552988 0.632611938388

1 个答案:

答案 0 :(得分:2)

您可以打开文件并写入文件(请参阅此Python tutorial):

with open("output.txt", "w") as fout:
    for i in range(528):
        for j in range(528):
             fout.write("%d %d %f %f %f %f\n" % (i,j,rho[i,j],rhop[i,j],ux[i,j],uy[i,j]))

或者,只需运行脚本并将输出传递给文件:

python script.py >output.txt