DEV Community

sunj
sunj

Posted on

Python, StringIO()

문자열을 파일객체처럼 만든다
데이터가 디스크에 기록되는 대신 메모리의 버퍼(문자열 버퍼)에 기록된다는 점을 제외하고는 일반 파일과 똑같이 사용할 수 있습니다

import csv
import io


def execute(f):
    result = []
    reader = csv.reader(f)
    for line in reader:
        one = int(line[0])
        two = int(line[1])
        three = one+two
        line.append(three)
        result.append(line)
    return result


src = '''\
20,40
50,90
77,22
'''

with io.StringIO(src) as f:  # 문자열을 파일객체럼 만든다.
    result = execute(f)
    print(result)
Enter fullscreen mode Exit fullscreen mode
[['20', '40', 60], ['50', '90', 140], ['77', '22', 99]]
Enter fullscreen mode Exit fullscreen mode

참조 : https://wikidocs.net/122776

Top comments (0)