將 for 循環每次迭代的輸出保存在單個文件中

Question

嗨專家我有一個包含 5 列數據的文本文件（input.txt），如下所示：

2 3 4 5 6
3 4 5 6 7
2 3 4 5 6 
6 7 8 9 0
3 3 3 3 3

我只想在每次迭代時讀取單列並想在其上添加 5 並希望 np.savetxt 將循環的每個迭代輸出保存在文本文件的不同列中

我寫了腳本

import numpy as np
import random


inpdata=np.loadtxt("input.txt")
for x in np.arange(0,4,1):
        data=inpdata[:,x]
        sum=data+5
        np.savetxt('tx_'+str(x),data)

在執行此輸出時保存在不同的文件中，但我需要將輸出保存在單個文件中作為第一列、第二列、第三列......我需要如下輸出

7  8  9  10  11
8  9  10 11  12
7  8  9  10  11 
11 12 13 14  5
8  8  8  8   8

我希望一些專家會幫助我。提前致謝。

Answer 1

因為您只是對數組執行一些操作。 您可以等效地執行以下操作：

1. 從txt文件中讀取數據
2. 執行您想要按列或任何其他方式執行的任何操作。
3. 將新數據保存到文件

在這個特定示例中，您可以向數組添加一個標量，如下所示：

import numpy as np
data=np.loadtxt("test.txt")
data += 5
np.savetxt('output.txt', data)

如果您堅持按列進行操作：

import numpy as np
data=np.loadtxt("test.txt")
for i in range(data.shape[1]):
    data[:, i] += 5
np.savetxt('output.txt', data)