【疾病研究部练习题一】

chenming

实际业务中，经常有需要读取excel文件的要求，例如在读取信息收集表的时候，同时也有时候需要把一些内容写入到excel文件当中，为了练习excel的读取，我们本周的题目是写一个脚本来读取excel文件中指定的一个或者多个sheet的内容并输出到一个文本文件中，使用\t进行分隔。用于读取的excel文件我给大家准备了一个，不过你也可以自己找一个。
大家可以在路径/WORK/Disease/chenming/Workshop/excel_example下找到该execl，同时你写的脚本也可以放在这个目录下，权限我已经修改了。
【大家可以在本帖下面回复自己的思路与我和其他小伙伴异同讨论】

chenming

之后的练习题我们将根据大家的反馈进行选择，大家有什么想实现的内容或者点子都可以在本帖下面回复

lmt

1. #!/usr/bin/env python
   
2. #-*- coding:utf-8 -*-
   
3. 
   
4. import sys
   
5. from openpyxl import load_workbook
   
6. 
   
7. Usage = "learn excel"
   
8. if len(sys.argv) < 3:
   
9.     print("Usage: learn how to read excel \npython exise.py workbook outfile ")
   
10.     exit()
    
11.    
    
12. ## open excel
    
13. def open_workbook(workbook):
    
14.     wb = load_workbook(workbook)
    
15.     sheet_names = wb.sheetnames
    
16.     print(sheet_names)
    
17.     dict_sheet_names = {}
    
18.     for index,name in enumerate(sheet_names):
    
19.         dict_sheet_names[index] = name
    
20.     return wb, dict_sheet_names
    
21. 
    
22. ## get needed sheet
    
23. def get_sheet(sheetname, wb, outfile):
    
24.     with open(outfile, 'w') as outdata:
    
25.         ws = wb[sheetname]     # get sheet by name
    
26.         for row in ws.rows:
    
27.             for cell in row:
    
28.                 outdata.write('%s\t' % cell.value)  
    
29.             outdata.write('\n')  
    
30. 
    
31. if __name__ == "__main__":
    
32.     workbook = sys.argv[1]
    
33.     outfile = sys.argv[2]
    
34.    
    
35.     wb, dict_sheet_names = open_workbook(workbook)
    
36.     for k,v in dict_sheet_names.items():
    
37.         print(k,":",v)
    
38.     sheetname = input('请输入想要查看的sheet名称：')
    
39.     get_sheet(sheetname,wb, outfile)

复制代码

chenming

lmt 发表于 2019-11-1 17:40

明祝的功能实现很棒，代码书写也较为规范，不过还可以再优化一些，例如，我如果事先并不知道sheet名称就没办法提取；sheet名称较长的时候，可不可以用数字来代替呢？

chenming

更正一下，是吕梦婷。。。

chenming

实际上，我们针对Excel2003版本，也就是以.xls为结尾的文件,我们需要使用xlrd和xlwt来处理，而针对Excel2007及以上的版本，则可以使用openpyxl模块来处理，这里我写了一个openpyxl实现的脚本，大家可以去我的github仓库查看https://github.com/KingCM/Utilit ... on/excel_to_text.py

chenming

关于使用openpyxl模块去操作一些excel的样式，比如字体，颜色，行宽，边宽和行高等内容，可以参考我写的一篇博客https://blog.csdn.net/qq_35696312/article/details/95060248

chenming

我们打开http://www.python-excel.org/之后可以看到一些可以使用的，使用python来操作excel的模块，每一个模块都对应了文档，我们对哪个模块感兴趣，就可以点击进去读一下。这里着重需要看的是openpyxl和xlrd以及xlwt三个模块

chenming

liyujie

1. #!usr/bin/env python2
   
2. #-*- coding: utf-8 -*-
   
3. 
   
4. from openpyxl import load_workbook
   
5. from openpyxl import Workbook
   
6. 
   
7. 
   
8. def sheetDes(wb):
   
9.     '''
   
10.     describe each sheet of excel
    
11.     sheetD={'sheetname':(maxrow,maxcolumn)}
    
12.     '''
    
13.     sheets=wb.get_sheet_names()
    
14.     sheetD={}
    
15.     for each in sheets:
    
16.         sheetD[each]=(wb.get_sheet_by_name(each).max_row,wb.get_sheet_by_name(each).max_column)
    
17.     return sheetD
    
18. 
    
19. def getSheetTitle(sn):
    
20.     '''
    
21.     get title of the sheetname you choose
    
22.     '''
    
23.     bs=wb.get_sheet_by_name(sn)
    
24.     for col in range(1, sheetD[sn][1]+1):
    
25.         val=bs.cell(1,col).value
    
26.         print col,val
    
27.     return bs
    
28. 
    
29. def getContext(cols,outfile):
    
30.     '''
    
31.     output columns to text
    
32.     '''
    
33.     with open(outfile,'w') as outf:
    
34.         for r in range(1, sheetD[sn][0]+1):
    
35.             line=''
    
36.             for c in cols.split(','):
    
37.                 line+=bs.cell(row=r, column=int(c)).value+'\t'
    
38.             outf.write(line.strip()+'\n')
    
39. 
    
40. if __name__=='__main__':
    
41.     wb = load_workbook('./AML_MDS_in_OMIM_and_HGMD.xlsx')
    
42.     sheetD=sheetDes(wb)
    
43.     for k,v in sheetD.items():
    
44.         print "sheet name:",k," rows:",v[0]," columns:",v[1]
    
45.     while True:
    
46.         sn=raw_input("please choose a sheetname:")
    
47.         if sn not in sheetD:
    
48.             sn=raw_input("please choose a sheetname:")
    
49.         else:
    
50.             bs=getSheetTitle(sn)
    
51.             tn=raw_input("titles split by comma:")
    
52.             getContext(tn,sn+'_'.join(tn.split(','))+'.txt')
    
53.             yn=raw_input("continue or exit (y/n):")
    
54.             if yn == "y":
    
55.                 continue
    
56.             elif yn == "n":
    
57.                 break
    
58.             else:
    
59.                 print "please choose y or n!"

复制代码