记一次教训,防止下次踩坑。
主要思路
- 按行读取文件,读出为字符串
- 利用json包转为字典格式
- 利用pandas包转为dataframe格式
- df格式直接to_excel为excel
具体代码
import json
import pandas as pd
'''
这个程序用来把companies.txt中的所有公司id和名字取出并导出
'''
f1 = open('companies.txt','r',encoding='utf-8')
keys = []
values = []
for s in f1.readlines():
dic = eval(s)
if keys == []:
keys = list(dic.keys())
values.append(list(dic.values()))
result_excel = pd.DataFrame(values, columns=keys) # 建立一个要写入excel的数据帧
result_excel = result_excel.set_index(keys[0]) # 把dataframe的索引(index)原来的0,1,2...去掉
result_excel.to_excel("./companies.xlsx") # 输出目录
f1.close()