不同形状数组Python的输出索引答案

【问题标题】：Out indices for different shaped arrays Python不同形状数组Python的输出索引
【发布时间】：2017-06-30 19:19:42
【问题描述】：

我正在尝试输出索引以及它的值。我可以在终端中打印它，但我想将它输入到 .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

【问题讨论】：

标签： python arrays numpy

【解决方案1】：

你可以打开一个文件并写入它（见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

【讨论】：