将多个numpy数组写入文件

Question

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

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

任何帮助表示赞赏。

谢谢！

Answer 1

回答一个非常古老的帖子供我自己使用。 我用以下内容写出了两个与CSV大小相同的1D阵列。

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')

Answer 2

如果您想将多个数组写入文件供以后使用，请查看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 3

import numpy


list1 = [1, 2, 3, 4]

list2 = [0.45, 0.98, 0.89, 0.21]

dat = numpy.array([list1, list2])

dat = dat.T

numpy.savetxt('data.txt', dat, delimiter = ',')