python比较两个excel表格的差异

efn阿克说

#!/usr/bin/python　　
#-*- coding:utf-8 -*-
　　

　　
import xlrd, os, sys
　　

　　
if len(sys.argv) != 3:
　　
  print "\n        ./script diff_1.xlsx diff_2.xlsx \n"
　　
  sys.exit()
　　

　　
diff = {}
　　
xls1 = sys.argv[1]
　　
xls2 = sys.argv[2]
　　

　　
if os.path.isfile(xls1) and os.path.isfile(xls2):
　　
  pass
　　
else:
　　
  print "\n        ./script diff_1.xlsx diff_2.xlsx \n"
　　
  sys.exit()
　　

　　
def getcontent(table):  #获取xlsx表内容
　　
  tmp_data = {}
　　
  tmp_table = ''
　　
  all_data = {}
　　

　　
  for j in xrange(table.nrows):
　　
    tmp = table.row_values(j)
　　
    if list(set(tmp)) == ['']:
　　
      tmp_data[tmp_table] = ''
　　
      if tmp_table != '':
　　
        all_data[j] = tmp_table
　　
      tmp_table = ''
　　
    else:
　　
      tmp2 = ""
　　
      for i in tmp:
　　
        try:
　　
          tmp2 = tmp2 + i + ","
　　
        except:
　　
          tmp2 = tmp2 + str(i)+ ","
　　
      tmp_table = tmp_table + tmp2 + "\n"  #把多行的内容放一起
　　
  return (tmp_data, all_data)
　　

　　
def write_file(excel_diff, filename):
　　
  f = open(filename, 'w')
　　
  f.write(excel_diff)
　　
  f.close()
　　

　　
def diff_content(table1,table2): #检查两个表差异
　　
  diff_tmp = []
　　
  for i in table1:
　　
    if i in table2:
　　
      pass
　　
    else:
　　
      diff_tmp.append(i)
　　
  return list(set(diff_tmp))
　　

　　
def get_rows(diff, all_data): #获取差异位置
　　
  tmp = []
　　
  for i in diff:
　　
    for j in all_data:
　　
      if all_data[j].strip() == i.strip():
　　
        tmp.append(j)
　　
        break
　　
  return tmp
　　

　　

　　
for i in range(0,2):  # 比较几个表
　　
  data1 = xlrd.open_workbook(xls1)
　　
  table1 = data1.sheets()
　　
  data2 = xlrd.open_workbook(xls2)
　　
  table2 = data2.sheets()
　　

　　
  tmp1, all1 = getcontent(table1)
　　
  tmp2, all2 = getcontent(table2)
　　

　　
  result = diff_content(tmp1, tmp2) #1 中有 2 没有的具体内容
　　
#  result2= diff_content(tmp2, tmp1)  #2 中有 1 没有的具体内容
　　

　　
  diff = get_rows(result,all1)
　　
  print sorted(diff)
　　

　　
#  write_file(sorted(diff), "/tmp/diff.txt")
　　

　　
#  diff = get_rows(result2,all2)
　　
#  print sorted(diff)