Python 如何将不同的数组一起写入文件

我想把某个hdf5文件中的一些信息写入一个新的txt文件。 信息包括时间、lat、lon、obs、协方差矩阵，其中每个变量都是数组

time

是1D数组，具有形状

（t，）

，

lat

和

lon

也是1D数组，具有形状

（t，）

，表示特定时间的位置，

obs

是2D数组，具有形状

（t，n）

，这意味着我们有t个观测，每个观测包含n个状态，协方差矩阵是一个3D数组，具有一个形状

（t，n，n）

。我想做的是逐个提取这些信息

with open('test.dat','w') as output_file:
    for i in np.arange(10):
        output_file.writelines(str(time[i])+' '+str(lat[i])+' '+str(lon[i])+'\n')
        output_file.writelines('%s ' % l for l in obs[i])
        output_file.write('\n')
        output_file.writelines('%s ' % l for l in cov[i])
        output_file.write('\n')

假设我们有

time=np.array（[1,2,3]）

，

lat=np.array（[20,30,40]）

，

lon=np.array（[40,50,70]）

，

obs=np.random.random（（3,4））

，

cov=np.random.random（（3,4,4））

， 我希望输出文件的写入方式如下：

time[0] lat[0] lon[0]
obs[0,0] obs[0,1] obs[0,2] obs[0,3] 
cov[0,0,0] cov[0,0,1] cov[0,0,2] cov[0,0,3]
....
....
cov[0,3,0] cov[0,3,1] cov[0,3,2] cov[0,3,3]
time[1] lat[1] lon[1]
.....

示例输出中的每一行都是输出文件中的一个文本行。一行中的字段用一个空格分隔。

---

但是我写的代码似乎不起作用，我该怎么办？

这段代码似乎起作用了。我使用

print（*anarray）

技巧轻松地将一维数组的内容打印在由空格分隔的一行中，从而避免了一些循环。我使用了

print

而不是

write

来轻松地在正确的位置获得行尾标记。我本可以省略

sep='

参数，因为它们是默认参数，但我认为“显式优于隐式”。我删除了

str

转换，因为

print

会自动进行转换。我还加入了一些

assert

语句来保证数组形状的一致性

import numpy as np

t = 3
n = 4

time = np.array([1, 2, 3])
lat = np.array([20, 30, 40])
lon = np.array([40, 50, 70])
obs = np.random.random((3, 4))
cov = np.random.random((3, 4, 4))

assert time.shape == lat.shape == lon.shape == (t,)
assert obs.shape == (t, n)
assert cov.shape == (t, n, n)

with open('test.dat', 'w') as output_file:
    for i in range(t):
        print(time[i], lat[i], lon[i], sep=' ', file=output_file)
        print(*obs[i], sep=' ', file=output_file)
        for j in range(n):
            print(*cov[i, j], sep=' ', file=output_file)

以下是运行该代码后文件

test.dat

的内容：

1 20 40
0.3299447219312376 0.996996139803863 0.9551526837239497 0.5509582503806248
0.8819189869595838 0.9132355846068761 0.9665353304879704 0.46635220924415366
0.5181587118344747 0.8038185825031468 0.6745351776394765 0.5995935654205414
0.8619757623942051 0.482577455518059 0.7957648986891931 0.7407148158441567
0.09795020526664455 0.794707302185401 0.5293100793000466 0.3338543543683751
2 30 50
0.8035406182773407 0.1185746504181443 0.580297768900217 0.18777926684714497
0.532529800187838 0.12522453075786388 0.46300433798756324 0.22510518678645997
0.29260759039257944 0.3045692764639927 0.8085267868489262 0.6155070434401326
0.744762931777478 0.3722679713441319 0.395290426225012 0.3838002215325691
0.8049013701055735 0.4221372661820496 0.9607451508389756 0.7143917783338681
3 40 70
0.747699155972642 0.0016320243955539881 0.6063364532047132 0.8264067710516358
0.5795303829414947 0.49210530688638277 0.7499924344997647 0.601564975827258
0.7578052140931559 0.3514264858000006 0.7016988143157379 0.7628124804393464
0.3503877412670806 0.4020137310035564 0.688387279488728 0.46935179605446187
0.21964712005015719 0.8918053594781274 0.7908691508919491 0.006491011595309404

---

您正在尝试使用HDF5吗？如果您执行

output\u file.write（'\n'）

，您将编写一个文本文件。我希望您希望使用h5py，并将h5py.File（'test.h5'）作为f:…而不是仅在file@RoryDaulton，抱歉，我已更正。@ScottStaniewicz，否，我想将hdf5中的不同数组放入新的txt文件。注意：

str[time[I]

将是无效语法，您可能需要

str（time[i]）

当使用

print（*anarray）

技巧时，我们是否可以以格式化的方式打印这些值，例如在

%e

@AllenZhang:考虑到工作方式，我看不到在使用

print（*anarray）

时如何格式化项目。这需要一个循环。循环可能隐藏在列表理解中，以生成格式化字符串列表，而不是浮点列表。