Numpy save 2d array to text file

I use

np.savetxt('file.txt', array, delimiter=',')

to save the array in a comma separated file. It looks like this:

 1, 2, 3 4, 5, 6 7, 8, 9 

How to save an array to a file shown in numpy format. In other words, it looks like this:

 [[1, 2, 3], [4, 5, 6], [7, 8, 9]] 
+7
python arrays numpy
source share
3 answers
 In [38]: x = np.arange(1,10).reshape(3,3) In [40]: print(np.array2string(x, separator=', ')) [[1, 2, 3], [4, 5, 6], [7, 8, 9]] 

To save a NumPy x array to a file:

 np.set_printoptions(threshold=np.inf, linewidth=np.inf) # turn off summarization, line-wrapping with open(path, 'w') as f: f.write(np.array2string(x, separator=', ')) 
+6
source share

You can also use the first format for copying:

 >>> from io import BytesIO >>> bio = BytesIO('''\ ... 1, 2, 3 ... 4, 5, 6 ... 7, 8, 9 ... ''') # copy pasted from above >>> xs = np.loadtxt(bio, delimiter=', ') >>> xs array([[ 1., 2., 3.], [ 4., 5., 6.], [ 7., 8., 9.]]) 
+3
source share
 import sys file = "<you_file_dir>file.txt" sys.stdout = open(file, 'w') d = [1,2,3,4,5,6,7,8,9] l__d1 = d[0:3] l__d2 = d[3:6] l__d3 = d[6:9] print str(l__d1) + '\n' + str(l__d2) + '\n' + str(l__d3) 
+2
source share

All Articles