python 编写输出到csv的操作_docker

这篇文章主要介绍了python 编写输出到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)

图片[1]-python 编写输出到csv的操作_docker-安全小天地

定义一个列表:然后将需要写的数据添加到列表

a #追加数据到下一行

newline=” //去除每一行中间的空行,若不加的话,行与行中间有空格

加newline

图片[2]-python 编写输出到csv的操作_docker-安全小天地

不加:

图片[3]-python 编写输出到csv的操作_docker-安全小天地

图片[4]-python 编写输出到csv的操作_docker-安全小天地

补充:利用python将json数据转换为csv格式

假设.json文件中存储的数据为:

{\"type\": \"Point\", \"link\": \"http://www.dianping.com/newhotel/22416995\", \"coordinates\": [116.37256372996957, 40.39798447055443], \"category\": \"经济型\", \"name\": \"北京荷塘山庄\", \"count\": \"278\", \"address\": \"北京市怀柔区黄花城村安四路\", \"price\": \"380\"}
{\"type\": \"Point\", \"link\": \"http://www.dianping.com/newhotel/19717653\", \"coordinates\": [116.56881588256466, 40.43310967948417], \"category\": \"经济型\", \"name\": \"慕田峪长城鱼师傅乡村酒店\", \"count\": \"89\", \"address\": \"北京市怀柔区渤海镇苇店村(慕田峪长城下3公里处,近怀黄路)\", \"price\": \"258\"}
{\"type\": \"Point\", \"link\": \"http://www.dianping.com/newhotel/58365289\", \"coordinates\": [116.62874974822378, 40.45610264855833], \"category\": \"经济型\", \"name\": \"北京蜜桃儿亲子客栈\", \"count\": \"119\", \"address\": \"北京市怀柔区神堂峪风景区下官地11号\", \"price\": \"549\"}

现在需要将上面的这些数据存为csv格式,其中字典的keys为csv中的属性名称,字典的values为csv中属性对应的值。

如果只需要按照json的keys来生成csv,那么操作比较简单,直接按照下面的方法即可:

#-*-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\') # 此处这样写会导致写出来的文件会有空行
  # 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) # 将属性列表写入csv中
      flag = False
    # 读取json数据的每一行,将values数据一次一行的写入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

其中第三个参数为需要转换的文件的路径和其名称,将其后缀删除。运行文件后即可得到转换后的csv文件。

如果需要对json文件中每个字典的key字段进行修改,比如需要将上面dict中的coordinate中的经纬度数据取出来存为x、y数据,则可以按照下面的方法(此方法还可以调整每个属性显示的顺序,效果更好一点):

import csv
import json
import sys
import codecs
def trans(path):
  jsonData = codecs.open(path+\'.json\', \'r\', \'utf-8\')
  # csvfile = open(path+\'.csv\', \'w\') # 此处这样写会导致写出来的文件会有空行
  # 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文件是我在大众点评抓取的数据,存储格式为utf-8。建议使用codecs包来读取json数据,可指定编码方式。

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

以上为个人经验,希望能给大家一个参考,也希望大家多多支持安全小天地。如有错误或未考虑完全的地方,望不吝赐教。

------本文已结束,感谢您的阅读------
THE END
喜欢就支持一下吧
点赞13 分享
评论 抢沙发

请登录后发表评论

    暂无评论内容