python - python: pandas + fpdf + double groupby
问题描述
我是 python 新手,但已经遇到了问题。
基于 .csv 表我想创建一个 pdf 文件
客户ID | item_id | 数量 | 项目价格 | 价格 | 仓储城 |
---|---|---|---|---|---|
1 | 1 | 1 | 10 | 10 | 一个 |
1 | 2 | 2 | 20 | 40 | 乙 |
1 | 3 | 1 | 30 | 30 | C |
1 | 4 | 1 | 40 | 40 | 一个 |
1 | 5 | 2 | 50 | 100 | 乙 |
2 | 1 | 3 | 10 | 30 | C |
2 | 2 | 1 | 20 | 20 | 一个 |
2 | 3 | 2 | 30 | 60 | 乙 |
3 | 1 | 2 | 10 | 20 | C |
3 | 2 | 1 | 20 | 20 | 一个 |
3 | 3 | 3 | 30 | 90 | 乙 |
3 | 1 | 1 | 10 | 10 | C |
对于每个 customer_id,我想提供一个页面,其中包含如下表格(例如 customer id_=1):
- 购买了哪些商品,它们的总价是多少
item_id | 数量 | 项目价格 | 价格 |
---|---|---|---|
1 | 1 | 10 | 10 |
2 | 2 | 20 | 40 |
3 | 1 | 30 | 30 |
4 | 1 | 40 | 40 |
5 | 2 | 50 | 100 |
-------- | -------- | -------- | -------- |
和 | 7 | 220 |
2. 有多少产品来自哪个仓库
仓储城 | 金额 |
---|---|
一个 | 2 |
乙 | 4 |
C | 1 |
-------- | -------- |
和 | 7 |
我尝试了以下解决方案,但我无法正确访问数据
import pandas as pd
import numpy as np
from fpdf import FPDF
class PDF(FPDF):
def header(self):
# Logo
#self.image('logo.png', 10, 8, 33)
# Arial bold 15
self.set_font('Arial', 'B', 15)
# Move to the right
self.cell(80)
# Title
self.cell(30, 10, 'Bill', 1, 0, 'C')
# Line break
self.ln(20)
# Page footer
def footer(self):
# Position at 1.5 cm from bottom
self.set_y(-15)
# Arial italic 8
self.set_font('Arial', 'I', 8)
# Page number
self.cell(0, 10, 'Page ' + str(self.page_no()) + '/{nb}', 0, 0, 'C')
def doc_header(self, num, label):
# Arial 12
self.set_font('Arial', '', 12)
# Background color
self.set_fill_color(200, 220, 255)
# Title
self.cell(0, 6, 'Customer number: %d - %s -' % (num, label), 0, 1, 'L', 1)
# Line break
self.ln(4)
def print_doc(self, num, title):
self.add_page()
self.doc_header(num, title)
### Read csv
ipl_data = pd.read_csv("test.csv", delimiter=";", header=0)
df = pd.DataFrame(ipl_data)
pdf = PDF()
df3 = df.groupby(['customer_id'])
df5 = df.groupby(['customer_id','StorageCity'])
for key,group_df in df3:
pdf.alias_nb_pages()
pdf.set_font('Times', '', 12)
page_width = pdf.w - 2 * pdf.l_margin
col_width = page_width/4
pdf.print_doc(key, 'test')
#print(group_df)
#print("the group for product '{}' has {} rows".format(key,len(group_df)))
pdf.ln(1)
th = pdf.font_size
pdf.cell(col_width, th, df.columns[1], border=1)
pdf.cell(col_width, th, df.columns[2], border=1)
pdf.cell(col_width, th, df.columns[3], border=1)
pdf.cell(col_width, th, df.columns[4], border=1)
pdf.ln(th)
for row_index, row in group_df.iterrows():
pdf.cell(col_width, th, format(row['item_id']), border=1)
pdf.cell(col_width, th, format(row['amount']), border=1)
pdf.cell(col_width, th, format(row['ItemPrice']), border=1)
pdf.cell(col_width, th, format(row['Price']), border=1)
pdf.ln(th)
pdf.ln(1)
pdf.ln(1)
for key,group_df in df5:
pdf.ln(1)
th = pdf.font_size
pdf.cell(col_width, th, df.columns[5], border=1)
pdf.cell(col_width, th, df.columns[1], border=1)
pdf.ln(th)
for row_index, row in group_df.iterrows():
pdf.cell(col_width, th, format(row['StorageCity']), border=1)
pdf.cell(col_width, th, format(row['item_id']), border=1)
pdf.ln(th)
pdf.ln(1)
pdf.ln(1)
pdf.output('test.pdf', 'F')
我将不胜感激任何提示。
解决方案
df=pd.read_csv("test.csv")
DF_list=list()
DF_list1=list()
for j in np.unique(df["customer_id"]):
tmp=df[df["customer_id"]==j][["item_id", "amount", "ItemPrice", "Price"]]
r=tmp.sum().tolist()
r[0]="Sum"
r[2]=""
tmp1=tmp
tmp1.loc[tmp1.shape[0]]=np.repeat("-----", 4)
tmp1.loc[tmp1.shape[0]]=r
DF_list.append(tmp1)
tmp2=df[df["customer_id"]==j][["StorageCity", "amount"]]
tmp2=tmp2.groupby("StorageCity").apply(sum)["amount"].reset_index(level=0).rename(columns={"amount":"SumAmount"})
r=tmp2.sum().tolist()
r[0]="Sum"
tmp2.loc[tmp2.shape[0]]=np.repeat("-----", 2)
tmp2.loc[tmp2.shape[0]]=r
DF_list1.append(tmp2)
pdf = PDF()
pdf.set_font('Times', '', 12)
page_width = pdf.w - 2 * pdf.l_margin
col_width = page_width/4
cust={0: "Bob", 1: "Smith", 2: "Sally"}
for i in range(len(DF_list)):
j=DF_list[i]
l=DF_list1[i]
pdf.print_doc(i+1, cust[i])
pdf.ln(1)
th = pdf.font_size
pdf.cell(col_width, th, j.columns[0], border=1)
pdf.cell(col_width, th, j.columns[1], border=1)
pdf.cell(col_width, th, j.columns[2], border=1)
pdf.cell(col_width, th, j.columns[3], border=1)
pdf.ln(th)
for row_index, row in j.iterrows():
pdf.cell(col_width, th, format(row['item_id']), border=1)
pdf.cell(col_width, th, format(row['amount']), border=1)
pdf.cell(col_width, th, format(row['ItemPrice']), border=1)
pdf.cell(col_width, th, format(row['Price']), border=1)
pdf.ln(th)
pdf.ln(1)
pdf.ln(1)
pdf.cell(col_width, th, l.columns[0], border=1)
pdf.cell(col_width, th, l.columns[1], border=1)
pdf.ln(th)
for row_index, row in l.iterrows():
pdf.cell(col_width, th, format(row['StorageCity']), border=1)
pdf.cell(col_width, th, format(row['SumAmount']), border=1)
pdf.ln(th)
pdf.ln(1)
推荐阅读
- android - 使用 Arraylist 填充房间数据库
- flutter - 如何在 Flutter 中加载时导航到另一个页面
- android - 如何通过 android 应用程序的整个布局移动自定义视图?
- java - 如何正确使用 Image.SCALE_DEFAULT?
- python - Python 类/对象 __str__ 和其他问题
- dart - Flutter 如何更新/添加值到映射中的键
- java - 使用对象数组对 Java 进行堆排序
- javascript - 使用 Javascript Entity Framework C# MVC Json 添加记录时出错
- javascript - 在 Hover 上仅从几个具有相同类的 Div 中更改一个
- java - 如何在java重载中获取结果值