import pysolr
import json
import pandas as pd
import os
#name 是文件名
#write_data参数data是json格式
class write_my_excel:
def __init__(self,name):
self.name=name+'.xlsx'
self.num=0
self.f=os.path.exists(self.name)
if self.f==False:
data = pd.DataFrame()
data.to_excel(self.name, sheet_name='sheet1',index=False)
self.writer = pd.ExcelWriter(self.name, mode='a',engine='openpyxl',if_sheet_exists='overlay')
self.get_cur_count()
def get_cur_count(self):
self.num=len(pd.read_excel(self.name))
return self.num
def write_data(self,data):
data=eval(data)
data = pd.DataFrame(data)
if self.num==0:
data.to_excel(excel_writer=self.writer, sheet_name='sheet1',index=False,startrow=self.num,header=True)
else:
data.to_excel(excel_writer=self.writer, sheet_name='sheet1',index=False,startrow=self.num,header=False)
self.num+=1
if self.num%1000==0:
self.writer.save()
def write_close(self):
self.writer.save()
self.writer.close()
#构造
w=write_my_excel('myfile1')
#写入
json_data=[{'id':1,'name':'张三'},\
{'id':2,'name':'李四'},\
{'id':3,'name':'王五','sex':'女'}]
json_data=json.dumps(json_data)
w.write_data(json_data)
#获取文件内数据数量(条)
print(w.get_cur_count())
#关闭
w.write_close()
02-26
1068
![](https://csdnimg.cn/release/blogv2/dist/pc/img/readCountWhite.png)
07-27
1万+
![](https://csdnimg.cn/release/blogv2/dist/pc/img/readCountWhite.png)
“相关推荐”对你有帮助么?
-
非常没帮助
-
没帮助
-
一般
-
有帮助
-
非常有帮助
提交