python - 使用 Pandas 将 Dataframe 行写入 Excel 工作表
问题描述
如何将返回的行从数据框中保存到 Excel 工作表中?
故事:我正在处理包含加拿大邮政编码的大型 txt 文件(1.7M 行)。我创建了一个数据框,并将我需要的值提取到其中。数据框的一列是省 id (df['PID'])
。我创建了一个在该 PID 列中找到的唯一值的列表,并在一个新的 Excel 电子表格中成功创建了 (13) 个工作表,每个工作表都以唯一的 PID 命名。
问题:每张表只包含标题,而不包含行的值。
我无法将匹配的行写入工作表。这是我的代码:
import pandas as pd
# parse text file into dataframe
path = 'the_file.txt'
df = pd.read_csv(path, sep='\t', header=None, names=['ORIG', 'PID','PCODE'], encoding='iso-8859-1')
# extract characters to fill values
df['ORIG'] = df['ORIG']
df['PID'] = df['ORIG'].str[11:13].astype(int)
df['PCODE'] = df['ORIG'].str[:6]
# create list of unique province ID's
prov_ids = df['PID'].unique().tolist()
prov_ids_string = map(str, prov_ids)
# create new excel file
writer = pd.ExcelWriter('CanData.xlsx', engine='xlsxwriter')
for id in prov_ids_string:
mydf = df.loc[df.PID==id]
# NEED TO WRITE VALUES FROM ROW INTO SHEET HERE*
mydf.to_excel(writer, sheet_name=id)
writer.save()
我知道写作应该发生在哪里,但我没有得到正确的结果。如何仅将具有匹配 PID 的行写入各自的工作表?
谢谢
解决方案
以下应该有效:
import pandas as pd
import xlsxwriter
# parse text file into dataframe
# extract characters to fill values
df['ORIG'] = df['ORIG']
df['PID'] = df['ORIG'].str[11:13].astype(int)
df['PCODE'] = df['ORIG'].str[:6]
# create list of unique province ID's
prov_ids = df['PID'].unique().tolist()
#prov_ids_string = map(str, prov_ids)
# create new excel file
writer = pd.ExcelWriter('./CanData.xlsx', engine='xlsxwriter')
for idx in prov_ids:
mydf = df.loc[df.PID==idx]
# NEED TO WRITE VALUES FROM ROW INTO SHEET HERE*
mydf.to_excel(writer, sheet_name=str(idx))
writer.save()
例如数据:
df = pd.DataFrame()
df['ORIG'] = ['aaaaaa111111111111111111111',
'bbbbbb2222222222222222222222']
df['ORIG'] = df['ORIG']
df['PID'] = df['ORIG'].str[11:13].astype(int)
df['PCODE'] = df['ORIG'].str[:6]
print(df)
氪。
推荐阅读
- php - 使用 .htaccess 在子文件夹中隐藏 .php
- javascript - 如何在尝试使用 JavaScript 获取用户在 Linkedin API 上的信息时修复“invalid://”消息
- office365 - 是否可以拦截 Office.js 上的事件?
- python - 绘图时,想要在 x 间隔内“保持”y 值。[不是“条形图”问题]
- go - 如何使用 redigo 将地图保存并检索到 redis 中?
- r - 如何计算所有成员对以进行网络分析?
- javascript - 如何使用 vanilla javascript 获取滚动开始事件
- angular - Angular Material:如何在材料表中制作多行列
- corda - 流量测试和账户
- python-3.x - Python OOP 继承用例和调用继承对象