파이썬 문자열, 배열을 txt, csv 파일로 출력, 저장하기

1. 텍스트를 txt 파일로 저장하기

 

line 4: 'w' 파일에 내용이 있으면 지워버리고 open부터 close까지의 내용만 파일로 저장한다.

line 5: 문자열 입력 후에 새 줄이 바로 시작되는 것이 아니어서 새로운 줄을 추가해주는 명령어를 입력하였다. 

a = "Adios."
b = "Hello."
 
f = open("file01.txt", 'w')
f.write(str(a)+"\n")
f.write("\n")
f.write(str(b))
 
f.close()

2. 행렬, 배열을 txt 형태로 저장하기/csv 형태로 저장하기

import numpy as np
a = np.mat([[1, 2],
              [3, 4]])
b = np.mat([[1, 5],
            [5, 1]])
 
np.savetxt('file02.csv', a, fmt='%d', delimiter=',')
np.savetxt('file03.txt', a+b, fmt='%d', delimiter=' ')

 

delimiter (구분 기호)는 공백이나 커마 등 문자열을 쓸 수 있다. 행렬의 칼럼 사이마다 이 문자열이 입력된다.

file02.csv 결과물

1,2
3,4

 

file03.txt의 결과물

2 7
8 5

line 8에서는 a+b를 하면 계산된 값이 출력되는지 궁금해서 해보았는데,

예상대로 행렬을 계산한 값이 텍스트 파일에 저장되었다.

 

구분 기호를 ','로 넣고 파일명을 csv로 지정해서 csv파일도 만들어지는 것을 확인하였다.

 

savetxt(header='#####', footer='#####')

header, footer도 넣을 수 있어서 엑셀에 셀마다 데이터가 들어가는 csv 파일로 만드는 데 쓰는 것도 가능할 것 같다.