python将文本转为excel

记一次教训，防止下次踩坑。

主要思路

1. 按行读取文件，读出为字符串
2. 利用json包转为字典格式
3. 利用pandas包转为dataframe格式
4. 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()