python 将同一借据号对应的两条紧急联系人信息组装成一条记录

Posted 2021-05-09

#!/usr/bin/env python3
#encoding:utf-8
import pandas as pd
import numpy as np

# 读取excel
df = pd.read_excel('m020171205.xlsx')

# 只保留第一项,并修改列名
df_first = df.drop_duplicates(['借据号'],keep='first')
df_first_rename = df_first.rename(
  columns={
    '紧急联系人电话': '紧急联系人1电话', 
    '紧急联系人姓名': '紧急联系人1姓名', 
    '紧急联系人关系': '紧急联系人1关系'
    }
  )

# 只保留第二项，并修改列名
df_last = df.drop_duplicates(['借据号'],keep='last')
df_last_rename = df_last.rename(
  columns={
    '紧急联系人电话': '紧急联系人2电话', 
    '紧急联系人姓名': '紧急联系人2姓名', 
    '紧急联系人关系': '紧急联系人2关系'
    }
  )

# merge合并，两个紧急联系人列名不同，不会合并
pd.merge(df_first_rename,df_last_rename)

# 生成xlsx表格
filename = 'm0'+ymd                                                          
writer = pd.ExcelWriter(filename+'.xlsx')                                    
df.to_excel(writer,filename,index=False)                                     
writer.save()