python 編寫(xiě)輸出到csv的操作

如下所示：

 def test_write(self):
 fields=[]
 fields.append(orderCode)
 with open(r'./test001.csv', 'a',newline="") as f:
  writer = csv.writer(f)
  writer.writerow(fields)

定義一個(gè)列表：然后將需要寫(xiě)的數據添加到列表

a #追加數據到下一行

newline='' //去除每一行中間的空行，若不加的話(huà)，行與行中間有空格

加newline

不加：

補充：利用python將json數據轉換為csv格式

假設.json文件中存儲的數據為：

{"type": "Point", "link": "http://www.dianping.com/newhotel/22416995", "coordinates": [116.37256372996957, 40.39798447055443], "category": "經(jīng)濟型", "name": "北京荷塘山莊", "count": "278", "address": "北京市懷柔區黃花城村安四路", "price": "380"}
{"type": "Point", "link": "http://www.dianping.com/newhotel/19717653", "coordinates": [116.56881588256466, 40.43310967948417], "category": "經(jīng)濟型", "name": "慕田峪長(cháng)城魚(yú)師傅鄉村酒店", "count": "89", "address": "北京市懷柔區渤海鎮葦店村(慕田峪長(cháng)城下3公里處，近懷黃路)", "price": "258"}
{"type": "Point", "link": "http://www.dianping.com/newhotel/58365289", "coordinates": [116.62874974822378, 40.45610264855833], "category": "經(jīng)濟型", "name": "北京蜜桃兒親子客棧", "count": "119", "address": "北京市懷柔區神堂峪風(fēng)景區下官地11號", "price": "549"}

現在需要將上面的這些數據存為csv格式，其中字典的keys為csv中的屬性名稱(chēng)，字典的values為csv中屬性對應的值。

如果只需要按照json的keys來(lái)生成csv，那么操作比較簡(jiǎn)單，直接按照下面的方法即可：

#-*-coding:utf-8-*-
import csv
import json
import sys
import codecs
def trans(path):
  jsonData = codecs.open(path+'.json', 'r', 'utf-8')
  # csvfile = open(path+'.csv', 'w') # 此處這樣寫(xiě)會(huì )導致寫(xiě)出來(lái)的文件會(huì )有空行
  # csvfile = open(path+'.csv', 'wb') # python2下
  csvfile = open(path+'.csv', 'w', newline='') # python3下
  writer = csv.writer(csvfile, delimiter='\t', quoting=csv.QUOTE_ALL)
  flag = True
  for line in jsonData:
    dic = json.loads(line[0:-1])
    if flag:
      # 獲取屬性列表
      keys = list(dic.keys())
      print (keys)
      writer.writerow(keys) # 將屬性列表寫(xiě)入csv中
      flag = False
    # 讀取json數據的每一行，將values數據一次一行的寫(xiě)入csv中
    writer.writerow(list(dic.values()))
  jsonData.close()
  csvfile.close()
if __name__ == '__main__':
  path=str(sys.argv[1]) # 獲取path參數
  print (path)
  trans(path)

在python3下運行，命令行輸入

python C:\Users\MaMQ\Documents\jsonToCsv.py C:\Users\MaMQ\Documents\data\geoFood

其中第三個(gè)參數為需要轉換的文件的路徑和其名稱(chēng)，將其后綴刪除。運行文件后即可得到轉換后的csv文件。

如果需要對json文件中每個(gè)字典的key字段進(jìn)行修改，比如需要將上面dict中的coordinate中的經(jīng)緯度數據取出來(lái)存為x、y數據，則可以按照下面的方法（此方法還可以調整每個(gè)屬性顯示的順序，效果更好一點(diǎn)）：

import csv
import json
import sys
import codecs
def trans(path):
  jsonData = codecs.open(path+'.json', 'r', 'utf-8')
  # csvfile = open(path+'.csv', 'w') # 此處這樣寫(xiě)會(huì )導致寫(xiě)出來(lái)的文件會(huì )有空行
  # csvfile = open(path+'.csv', 'wb') # python2下
  csvfile = open(path+'.csv', 'w', newline='') # python3下
  writer = csv.writer(csvfile, delimiter='\t', quoting=csv.QUOTE_ALL)
  keys=['id', 'name', 'category', 'price', 'count', 'type', 'address', 'link', 'x', 'y']
  writer.writerow(keys)
  i = 1
  for dic in jsonData:
    dic = json.loads(dic[0:-1])
    x = dic['coordinates'][0]
    y = dic['coordinates'][1]
    writer.writerow([str(i),dic['name'],dic['category'],dic['price'],dic['count'],dic['type'],dic['address'],dic['link'],x,y])
    i += 1
  jsonData.close()
  csvfile.close()
if __name__ == '__main__':
  path = str(sys.argv[1])
  print (path)
  trans(path)

運行方法同上。

json文件是我在大眾點(diǎn)評抓取的數據，存儲格式為utf-8。建議使用codecs包來(lái)讀取json數據，可指定編碼方式。

jsonData = codecs.open(path + '.json', 'r', encoding='utf-8')

以上為個(gè)人經(jīng)驗，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。如有錯誤或未考慮完全的地方，望不吝賜教。