python怎么在csv修改数据
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) |
定义一个列表:然后将需要写的数据添加到列表
a #追加数据到下一行
newline='' //去除每一行中间的空行,若不加的话,行与行中间有空格
加newline
不加:
补充:利用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' ) |
以上为个人经验,希望能给大家一个参考,也希望大家多多支持开心学习网。如有错误或未考虑完全的地方,望不吝赐教。
原文链接:https://blog.csdn.net/qq_43422918/article/details/92795804