星期三, 4月 11, 2012

csv與StringIO

看了 python 網站上對 csv 的說明,就在想是不是一定只能藉著開檔才能使用呢?後來查了一下,然後看到 StringIO,藉著這個,就可以把字串以類似檔案的方式來操作了。
import csv
from StringIO import StringIO

def parse_to_get_record( csv_content ):
    """
    According to csv to parse result.
    """
    csv_lines = csv_content.splitlines()
    # 還可以濾掉某些不要的行
    csv_reader = csv.reader( StringIO( '\n'.join(csv_lines[2:-2]) ) )
    result = []
    for row in csv_reader:
        print( row )
        result.append( row )
    return result

f = open('d:\\tmp\\aaa.csv', 'r')
s = f.read()
print( parse_to_get_record(s) )

沒有留言: