2022年 11月 6日

python读取excel表格数据

python操作excel主要用到xlrd和xlwt两个库，xlrd读取表格数据，支持xlsx和xls格式的excel表格；xlwt写入excel表格数据

一、python读取excel表格数据

1、读取excel表格数据常用操作

  1. import xlrd
  2.  
  3. # 打开excel表格
  4. data_excel = xlrd.open_workbook('data/dataset.xlsx')
  5.  
  6.  
  7. # 获取所有sheet名称
  8. names = data_excel.sheet_names()
  9.  
  10.  
  11. # 获取book中的sheet工作表的三种方法,返回一个xlrd.sheet.Sheet()对象
  12. table = data_excel.sheets()[0]     # 通过索引顺序获取sheet
  13. table = data_excel.sheet_by_index(sheetx=0)     # 通过索引顺序获取sheet
  14. table = data_excel.sheet_by_name(sheet_name='Sheet1')    # 通过名称获取
  15.  
  16.  
  17. # excel工作表的行列操作
  18. n_rows = table.nrows    # 获取该sheet中的有效行数
  19. n_cols = table.ncols    # 获取该sheet中的有效列数
  20. row_list = table.row(rowx=0)    # 返回某行中所有的单元格对象组成的列表
  21. cols_list = table.col(colx=0)    # 返回某列中所有的单元格对象组成的列表
  22.  
  23.  
  24. # 返回某行中所有单元格的数据组成的列表
  25. row_data=table.row_values(0,start_colx=0,end_colx=None)
  26.  
  27.  
  28. # 返回某列中所有单元格的数据组成的列表
  29. cols_data=table.col_values(0,start_rowx=0,end_rowx=None)
  30. row_lenth=table.row_len(0)   # 返回某行的有效单元格长度
  31.  
  32.  
  33. # excel工作表的单元格操作
  34. row_col=table.cell(rowx=0,colx=0) # 返回单元格对象
  35. row_col_data=table.cell_value(rowx=0,colx=0) # 返回单元格中的数据

2、xlrd模块主要操作

  1. import xlrd
  2.  
  3.   
  4. """ 打开excel表格 """
  5. workbook = xlrd.open_workbook("data.xlsx")
  6. print(workbook)       # 结果：<xlrd.book.Book object at 0x000000000291B128>
  7.  
  8.  
  9. """ 获取所有sheet名称 """
  10. sheet_names = workbook.sheet_names()
  11. print(sheet_names)     # 结果：['表1', 'Sheet2']
  12.  
  13.  
  14. """ 获取所有或某个sheet对象 """
  15. # 获取所有的sheet对象
  16. sheets_object = workbook.sheets()
  17. print(sheets_object)    # 结果：[<xlrd.sheet.Sheet object at 0x0000000002956710>,<xlrd.sheet.Sheet object at 0x0000000002956AC8>]
  18.  
  19.  
  20. # 通过index获取第一个sheet对象
  21. sheet1_object = workbook.sheet_by_index(0)
  22. print(sheet1_object)    # 结果：<xlrd.sheet.Sheet object at 0x0000000002956710>
  23.  
  24.  
  25. # 通过name获取第一个sheet对象
  26. sheet1_object = workbook.sheet_by_name(sheet_name="表1")
  27. print(sheet1_object)    # 结果：<xlrd.sheet.Sheet object at 0x0000000002956710>
  28.   
  29.  
  30. """ 判断某个sheet 是否已导入"""
  31. # 通过index判断sheet1是否导入
  32. sheet1_is_load = workbook.sheet_loaded(sheet_name_or_index=0)
  33. print(sheet1_is_load)    # 结果：True
  34.  
  35.  
  36. # 通过sheet名称判断sheet1是否导入
  37. sheet1_is_load = workbook.sheet_loaded(sheet_name_or_index="表1")
  38. print(sheet1_is_load)    # 结果：True
  39.   
  40.  
  41. """ 对sheet对象中的行执行操作 """
  42. # 获取sheet1中的有效行数
  43. nrows = sheet1_object.nrows
  44. print(nrows)        # 结果：5
  45.  
  46.  
  47. # 获取sheet1中第3行的数据
  48. all_row_values = sheet1_object.row_values(rowx=2)
  49. print(all_row_values)      # 结果：[3.0, 'b', 1, '']
  50.  
  51. row_values = sheet1_object.row_values(rowx=2, start_colx=1, end_colx=3)
  52. print(row_values)        # 结果：['b', 1]
  53.  
  54.  
  55. # 获取sheet1中第3行的单元对象
  56. row_object = sheet1_object.row(rowx=2)
  57. print(row_object)        # 结果：[number:3.0, text:'b', bool:1, empty:'']
  58.  
  59.  
  60. # 获取sheet1中第3行的单元
  61. row_slice = sheet1_object.row_slice(rowx=2)
  62. print(row_slice)        # 结果：[number:3.0, text:'b', bool:1, empty:'']
  63.  
  64.  
  65. # 获取sheet1中第3行的单元类型
  66. row_types = sheet1_object.row_types(rowx=2)
  67. print(row_types)        # 结果：array('B', [2, 1, 4, 0])
  68.  
  69.  
  70. # 获取sheet1中第3行的长度
  71. row_len = sheet1_object.row_len(rowx=2)
  72. print(row_len)         # 结果：4
  73.  
  74.  
  75. # 获取sheet1所有行的生成器
  76. rows_generator = sheet1_object.get_rows()
  77. print(rows_generator)      # 结果：<generator object Sheet.get_rows.<locals>.<genexpr> at 0x00000000028D8BA0>
  78.   
  79.  
  80. """ 对sheet对象中的列执行操作 """
  81. # 获取sheet1中的有效列数
  82. ncols = sheet1_object.ncols
  83. print(ncols)        # 结果：4
  84.  
  85.  
  86. # 获取sheet1中第colx=1列的数据
  87. col_values = sheet1_object.col_values(colx=1)
  88. print(col_values)      # 结果：['测试', 'a', 'b', 'c', 'd']
  89.  
  90. col_values1 = sheet1_object.col_values(1, 1, 3)
  91. print(col_values1)     # 结果：['a', 'b']
  92.  
  93.  
  94. # 获取sheet1中第2列的单元
  95. col_slice = sheet1_object.col_slice(colx=1)
  96. print(col_slice)      # 结果：[text:'测试', text:'a', text:'b', text:'c', text:'d']
  97.  
  98.  
  99. # 获取sheet1中第2列的单元类型
  100. col_types = sheet1_object.col_types(colx=1)
  101. print(col_types)      # 结果：[1, 1, 1, 1, 1]
  102.   
  103.  
  104. """对sheet对象中的单元执行操作"""
  105. # 获取sheet1中第rowx=1行，第colx=2列的单元对象
  106. cell_info = sheet1_object.cell(rowx=1, colx=2)
  107. print(cell_info)      # 结果: text:'m'
  108. print(type(cell_info))   # 结果：<class 'xlrd.sheet.Cell'>
  109.  
  110.  
  111. # 获取sheet1中第rowx=1行，第colx=2列的单元值
  112. cell_value = sheet1_object.cell_value(rowx=1, colx=2)
  113. print(cell_value)      # 结果: m
  114.  
  115.  
  116. # 获取sheet1中第rowx=1行，第colx=2列的单元类型值
  117. cell_type = sheet1_object.cell_type(rowx=1, colx=2)
  118. print(cell_type)      # 结果：1
  119.   
  120. #单元类型ctype：empty为0，string为1，number为2，date为3，boolean为4，error为5；

 3、读取单元格内容为日期时间的方式

  • 若单元格内容的类型为date，即ctype值为3时，则代表此单元格的数据为日期
  • xlrd.xldate_as_tuple(xldate, datemode)：若xldate数据为日期/时间，则将转化为适用于datetime的元组 ， 返回值为元组，格式为：(year, month, day, hour, minute, nearest_second)
  • xldate：sheet对象中单元格的数据
  • datemode：日期模式
  1. import xlrd
  2. import datetime
  3.   
  4.  
  5. """ 读取sheet对象中的日期 """
  6. workbook = xlrd.open_workbook("data.xlsx")
  7. sheet2_object = workbook.sheet_by_name("Sheet2")
  8.  
  9.  
  10. # value_type = sheet2_object.cell(0, 1).ctype
  11. value_type = sheet2_object.cell_type(0, 1)
  12. print(value_type) # 结果：3 ，表示该值为date
  13.  
  14.  
  15. if value_type == 3:
  16.   print("单元格数据为日期")
  17.   cell_value = sheet2_object.cell_value(1, 0)
  18.   print(cell_value) # 结果：43567.0
  19.  
  20.   date_tuple = xlrd.xldate_as_tuple(cell_value, workbook.datemode)
  21.   print(date_tuple) # 结果：(2020, 4, 12, 0, 0, 0)
  22.  
  23.   date_value = datetime.date(*date_tuple[:3])
  24.   print(date_value) # 结果：2020-04-12
  25.  
  26.   date_format = date_value.strftime('%Y/%m/%d')
  27.   print(date_format) # 结果：2020/04/12

4、读取合并单元格的数据

若表格为xls格式的，打开workbook时需将formatting_info设置为True，然后再获取sheet中的合并单元格；若表格有xlsx格式的，打开workbook时保持formatting_info为默认值False，然后再获取sheet中的合并单元格；

SheetObject.merged_cells：获取sheet中合并单元格的信息，返回值为列表；若sheet对象中无合并单元格，则返回值为空列表；列表中每个单元格信息的格式为：(row_start, row_end, col_start, col_end)； row_start表示合并单元格的起始行； row_end表示合并单元格的结束行； col_start表示合并单元格的起始列；col_end表示合并单元格的结束列；合并单元格的行取值范围为[row_start, row_end)，包括row_start，不包括row_end；合并单元格的列取值范围为[col_start, col_end)，包括col_start，不包括col_end；如：(1, 3, 4, 6)：表示从第1到2行合并，从第4到第5列合并；

读取合并单元格数据仅需merged_cells数据中的row_start和col_start这两个索引即可

  1. import xlrd
  2.  
  3.   
  4. """ 获取合并的单元格并读取单元格数据 """
  5. # 获取xlsx格式的excel文件中的合并单元格
  6. workbook = xlrd.open_workbook("data.xlsx")
  7. sheet2_object = workbook.sheet_by_name("Sheet2")
  8. print(sheet2_object.merged_cells)  # 结果: [(1, 2, 0, 2), (3, 6, 0, 2)]
  9.  
  10.  
  11. # 获取xls格式的excel文件中的合并单元格
  12. workbook1 = xlrd.open_workbook("data.xls", formatting_info=True)
  13. sheet2_object1 = workbook1.sheet_by_name("Sheet2")
  14. print(sheet2_object1.merged_cells)  # 结果: [(1, 2, 0, 2), (3, 6, 0, 2)]
  15.   
  16.  
  17. # 读取合并单元格数据（仅需“起始行起始列”即可获取数据）
  18. print(sheet2_object.cell_value(1, 0))  # 结果：总结1
  19. print(sheet2_object.cell_value(3, 0))  # 结果：总结2
  20.  
  21.  
  22. # 或使用for循环获取所有的合并单元格数据
  23. for (row_start, row_end, col_start, col_end) in sheet2_object.merged_cells:
  24.     print(sheet2_object.cell_value(rowx=row_start, colx=col_start))

 二、python写入excel表格数据

1、写入excel表格数据常用操作和格式设置

  1. import xlwt
  2.   import datetime
  3.   # 创建一个workbook 设置编码
  4.   workbook = xlwt.Workbook(encoding='utf-8')
  5.  
  6.  
  7.   # 创建一个worksheet
  8.   worksheet = workbook.add_sheet('Sheet1')
  9.   
  10.  
  11.   #字体样式设置
  12.   style = xlwt.XFStyle()  # 初始化样式
  13.   font = xlwt.Font()  # 为样式创建字体
  14.   font.name = 'Times New Roman'
  15.   font.height = 20 * 11  # 字体大小，11为字号，20为衡量单位
  16.   font.bold = True  # 黑体
  17.   font.underline = True # 下划线
  18.   font.italic = True # 斜体字
  19.   style.font = font # 设定样式
  20.   
  21.  
  22.   # 数据写入excel,参数对应 行, 列, 值
  23.   worksheet.write(0, 0, 'test_data') # 不带样式的写入
  24.   worksheet.write(1, 0, 'test_data', style) # 带字体样式的写入
  25.   
  26.  
  27.   # 设置单元格宽度
  28.   worksheet.col(0).width = 3333
  29.   
  30.  
  31.   #设置单元格背景颜色
  32.   pattern = xlwt.Pattern()
  33.   pattern.pattern = xlwt.Pattern.SOLID_PATTERN
  34.   pattern.pattern_fore_colour = 13
  35.   style = xlwt.XFStyle() # Create the Pattern
  36.   style.pattern = pattern # Add Pattern to Style
  37.   worksheet.write(2, 0, 'colour', style)
  38.   
  39.  
  40.   #给单元格添加边框方法一
  41.   borders = xlwt.Borders() # Create Borders
  42.   borders.left = xlwt.Borders.DASHED  #DASHED虚线，NO_LINE没有，THIN实线
  43.   borders.right = xlwt.Borders.DASHED #borders.right=1 表示实线
  44.   borders.top = xlwt.Borders.DASHED
  45.   borders.bottom = xlwt.Borders.DASHED
  46.   borders.left_colour=0x40
  47.   borders.right_colour = 0x40
  48.   borders.top_colour = 0x40
  49.   borders.bottom_colour = 0x40
  50.   style = xlwt.XFStyle() # Create Style
  51.   style.borders = borders # Add Borders to Style
  52.   worksheet.write(3,0 , 'border1', style)
  53.   
  54.  
  55.   #给单元格添加边框方法二
  56.   # 细实线:1，小粗实线:2，细虚线:3，中细虚线:4，大粗实线:5，双线:6，细点虚线:7，大粗虚线:8，细点划线:9，粗点划线:10，细双点划线:11，粗双点划线:12，斜点划线:13
  57.   borders = xlwt.Borders()
  58.   borders.left = 1  #设置为细实线
  59.   borders.right = 1
  60.   borders.top = 1
  61.   borders.bottom = 1
  62.   borders.left_colour = 2 #颜色设置为红色
  63.   borders.right_colour = 2
  64.   borders.top_colour = 2
  65.   borders.bottom_colour = 2
  66.   style = xlwt.XFStyle() # Create Style
  67.   style.borders = borders # Add Borders to Style
  68.   worksheet.write(4, 0, 'border2', style)
  69.   
  70.  
  71.   #输入一个日期到单元格
  72.   style = xlwt.XFStyle()
  73.   style.num_format_str = 'M/D/YY' # Other options: D-MMM-YY, D-MMM, MMM-YY, h:mm, h:mm:ss, h:mm, h:mm:ss, M/D/YY h:mm, mm:ss, [h]:mm:ss, mm:ss.0
  74.   worksheet.write(5, 0, datetime.datetime.now(), style)
  75.   
  76.  
  77.   #单元格添加计算公式
  78.   worksheet.write(0, 1, 2) # Outputs 2
  79.   worksheet.write(0, 2, 3) # Outputs 3
  80.   worksheet.write(1, 1, xlwt.Formula('B1*C1')) # Should output "6" (B1[2] * B2[6])
  81.   worksheet.write(1, 2, xlwt.Formula('SUM(B1,C1)')) # Should output "5" (B1[2] + C1[3])
  82.   
  83.  
  84.   #向单元格添加一个超链接
  85.   worksheet.write(0, 3, xlwt.Formula('HYPERLINK("http://www.baidu.com";"baidu")')) # Outputs the text "baidu" linking to http://www.baidu.com
  86.   
  87.  
  88.   #单元格合并
  89.   worksheet.write_merge(0, 0, 4, 5, 'First Merge')  #合并0行的4到5列
  90.   worksheet.write_merge(1, 2, 4, 5, 'Second Merge') #合并1和2行的4到5列
  91.   
  92.  
  93.   #设置单元格内容的对其方式
  94.   alignment=xlwt.Alignment() ## Create Alignment
  95.   alignment.horz=xlwt.Alignment.HORZ_CENTER
  96.   alignment.vert=xlwt.Alignment.VERT_CENTER
  97.   style=xlwt.XFStyle()
  98.   style.alignment=alignment  # Add Alignment to Style
  99.   worksheet.write(0, 6, 'alignment', style)
  100.   
  101.  
  102.   # 保存文件
  103.   workbook.save('data_test.xls')

2、字体颜色和背景颜色对应索引号字体颜色：font.colour_index背景颜色：pattern.pattern_fore_colour