1. 程式人生 > >python 儲存資料為excel格式和txt格式

python 儲存資料為excel格式和txt格式

 excel 儲存方法:

book = xlwt.Workbook() 
#建立表單
sheet = book.add_sheet(u'sheet1',cell_overwrite_ok=True)

sheet.write(0,0,'id')
sheet.write(0,1,'text')
sheet.write(0,2,'user_id')
sheet.write(0,3,'geo_coordinates1')
sheet.write(0,4,'geo_coordinates2')
sheet.write(0,5,'created_at')
i =1
for all in searchRes:
    sheet.write(i,0,all['_id'])
    sheet.write(i,1,all['text'])
    sheet.write(i,2,all['user_id'])
    sheet.write(i,3,all['geo']['coordinates'][0])
    sheet.write(i,4,all['geo']['coordinates'][1])
    sheet.write(i,5,all['created_at'])
    i=i+1
    if(i==65530):
        break
book.save('Excel_Workbook.xls')

workshop 的寫入方法十分簡潔:使用write函式

            write(i,j,data)    ##i,j分別為excel的i行j列的位置。起始為0,0

            最後記得使用save

注意:excel由於限制最多存 65536行資料,因此面對海量的資料我們使用文字txt儲存

result如下。

txt儲存方法:

f = open('test1.txt','w',encoding='utf-8')
f.write('id')
f.write('\t')
f.write('text')
f.write('\t')
f.write('user_id')
f.write('\t')
f.write('geo_coordinates1')
f.write('\t')
f.write('geo_coordinates2')
f.write('\t')
f.write('created_at')
f.write('\n')
for all in searchRes:
    f.write(str(all['_id']))
    f.write('\t')
    f.write(all['text'])
    f.write('\t')
    f.write(str(all['user_id']))
    f.write('\t')
    f.write(str(all['geo']['coordinates'][0]))
    f.write('\t')
    f.write(str(all['geo']['coordinates'][1]))
    f.write('\t')
    f.write(all['created_at'])
    f.write('\n')
f.close()

比excel更為簡潔。直接上程式碼如上。