CSV格式的读写
import csv
def read_csv():
with open('test.csv','r',encoding='utf8') as csvFile:
read = csv.reader(csvFile)
for i in read:
print(i)
def read_csv_row():
with open('test.csv','r',encoding='utf8') as csvFile:
read = csv.reader(csvFile)
for (row,v) in enumerate(read):
if(row>0):
print('%d -> %s'%(row,v))
商品列表写入
productList = [{
"sku": 1,
"name": "iphone 12 pro max",
"price": 12999,
"brand": "苹果"
}, {
"sku": 2,
"name": "mate 40",
"price": 7999,
"brand": "华为"
}, {
"sku": 3,
"name": "vivo r9",
"price": 3999,
"brand": "vivo"
}]
def write_csv_list(list):
list_new = []
print(list)
for obj in list:
list_new.append([
obj['sku'],
obj['name'],
obj['price'],
obj['brand']
])
print(list_new)
with open('test.csv','w',encoding='utf8',newline='') as csvFile:
writer = csv.writer(csvFile)
writer.writerow(['型号','名字','价格','品牌'])
writer.writerows(list_new)
字典方式写入
def write_csv_dic(list):
with open('test.csv', 'w', encoding='utf8', newline='') as csvFile:
writer = csv.DictWriter(csvFile,['sku','name','price','brand'])
writer.writeheader()
writer.writerows(list)
write_csv_dic(productList)
def csv_write():
with open("test.csv",mode='w',encoding='utf-8') as file:
file.write('id,name,age')
file.write('\n')
file.write('1,张三,18')
file.write('\n')
file.write('2,李四,19')
file.write('\n')
file.write('3,王五,16')
csv_write()
自定义分隔符
import csv
with open('test.csv','r',encoding='utf8') as csvfile:
liness = csv.reader(csvfile,delimiter=';')
for i,v in enumerate(liness):
print(v[2])
if i > 1:
break
liness = csv.reader(csvfile)
for i, v in enumerate(liness):
print( v[2])