将多个numpy数组写入文件

Question

我知道如何使用numpy.savetxt将数组写入文件.如何将多个数组写入同一个文件？

基本上我想对一列数字进行数学运算,然后用修改后的数字替换旧列.我读到最简单的方法是完全编写一个新文件,将修改后的数字放入,然后"复制并粘贴"文件中的其他数字.

任何帮助表示赞赏.

谢谢!

Answer 1

如果您想将多个数组写入文件供以后使用,请查看numpy.savez.

但是,根据您的描述,听起来您想要对分隔文本文件的特定列执行某些操作.

在这种情况下,只需加载整个内容并仅对您需要的列进行操作.

例如

import numpy as np

data = np.loadtxt('test.txt')

# Multiply the 4th column by 5
data[:,3] *= 5

# Do something more complicated to the 2nd column
data[:,1] = np.cos(data[:,1])

# Save the array back to the file
np.savetxt('test.txt', data)

Answer 2

回答一个非常老的帖子供我自己使用。我使用以下内容写出了两个大小与CSV相同的一维数组。

import numpy as np

x = [1, 2, 3]
y = [4, 5, 6]
zipped = zip(x, y)
# >>> [(1, 4), (2, 5), (3, 6)]

# Save the array back to the file
np.savetxt('z.csv', zipped, fmt='%i,%i')