将循环中的数组保存到列中的一个txt文件中

Question

I'm creating an array of zeros (587x1) in which I want to replace a zero with a one in a specific line of the array given as an index from another file. This part works fine in my code so far.

Afterwards, I want to save all these newly created arrays in one txt file as columns next to each other, separated by tabs. What my code does, however, is overwriting the arrays. How do I manage to append them next to each other in one file?

Thanks a lot for your help!

Update: I managed to write all the arrays to one file, however, they are now simply printed on top of each other - how do I write them to columns appearing next to each other?

import os
import numpy as np
participants = ['001']
for vp in participants:
   with open('file.txt') as f:
      content = f.readlines()
   content = [x.strip() for x in content]
   content = map(int, content)
f = open(outfile.txt, 'w')       
for line in content:
      with open('outfile.txt', 'a') as f:
            arr = np.zeros((587, 1), dtype = int)
            np.put(arr, [line], [1])
            np.savetxt(f, arr, fmt='%i')
f.close()

Answer 1

You can try this code, that adds a given (n,1) shaped array as a column to a textfile that contains a (n,m) shaped matrix of integers:

def appendAsColumn(arr):
    fileContent = np.loadtxt('outfile.txt', dtype = int, ndmin = 2)
    fileContent = np.hstack((fileContent, arr.astype(int)))
    np.savetxt('outfile.txt', fileContent, fmt='%i')

Note that this will not work for the very first entry of the file, you will need to call

np.savetxt('outfile.txt', arr, fmt='%i')

for the first column.