zoukankan      html  css  js  c++  java
  • 【转发】Python使用openpyxl读写excel文件

    Python使用openpyxl读写excel文件

    这是一个第三方库,可以处理xlsx格式的Excel文件。pip install openpyxl安装。如果使用Aanconda,应该自带了。

    读取Excel文件

    需要导入相关函数。

    1. from openpyxl import load_workbook
    2. # 默认可读写,若有需要可以指定write_only和read_only为True
    3. wb = load_workbook('mainbuilding33.xlsx')

    默认打开的文件为可读写,若有需要可以指定参数read_onlyTrue

    获取工作表--Sheet

    1. # 获得所有sheet的名称
    2. print(wb.get_sheet_names())
    3. # 根据sheet名字获得sheet
    4. a_sheet = wb.get_sheet_by_name('Sheet1')
    5. # 获得sheet名
    6. print(a_sheet.title)
    7. # 获得当前正在显示的sheet, 也可以用wb.get_active_sheet()
    8. sheet = wb.active

    获取单元格

    1. # 获取某个单元格的值,观察excel发现也是先字母再数字的顺序,即先列再行
    2. b4 = sheet['B4']
    3. # 分别返回
    4. print(f'({b4.column}, {b4.row}) is {b4.value}') # 返回的数字就是int型
    5. # 除了用下标的方式获得,还可以用cell函数, 换成数字,这个表示B4
    6. b4_too = sheet.cell(row=4, column=2)
    7. print(b4_too.value)

    b4.column返回Bb4.row返回4, value则是那个单元格的值。另外cell还有一个属性coordinate, 像b4这个单元格返回的是坐标B4

    获得最大行和最大列

    1. # 获得最大列和最大行
    2. print(sheet.max_row)
    3. print(sheet.max_column)

    获取行和列

    • sheet.rows为生成器, 里面是每一行的数据,每一行又由一个tuple包裹。
    • sheet.columns类似,不过里面是每个tuple是每一列的单元格。
    1. # 因为按行,所以返回A1, B1, C1这样的顺序
    2. for row in sheet.rows:
    3. for cell in row:
    4. print(cell.value)
    5. # A1, A2, A3这样的顺序
    6. for column in sheet.columns:
    7. for cell in column:
    8. print(cell.value)

    上面的代码就可以获得所有单元格的数据。如果要获得某行的数据呢?给其一个索引就行了,因为sheet.rows是生成器类型,不能使用索引,转换成list之后再使用索引,list(sheet.rows)[2]这样就获取到第三行的tuple对象。

    1. for cell in list(sheet.rows)[2]:
    2. print(cell.value)

    如何获得任意区间的单元格?

    可以使用range函数,下面的写法,获得了以A1为左上角,B3为右下角矩形区域的所有单元格。注意range从1开始的,因为在openpyxl中为了和Excel中的表达方式一致,并不和编程语言的习惯以0表示第一个值。

    1. for i in range(1, 4):
    2. for j in range(1, 3):
    3. print(sheet.cell(row=i, column=j))
    4. # out
    5. <Cell mainbuilding33.A1>
    6. <Cell mainbuilding33.B1>
    7. <Cell mainbuilding33.A2>
    8. <Cell mainbuilding33.B2>
    9. <Cell mainbuilding33.A3>
    10. <Cell mainbuilding33.B3>

    还可以像使用切片那样使用。sheet['A1':'B3']返回一个tuple,该元组内部还是元组,由每行的单元格构成一个元组。

    1. for row_cell in sheet['A1':'B3']:
    2. for cell in row_cell:
    3. print(cell)
    4. for cell in sheet['A1':'B3']:
    5. print(cell)
    6. # out
    7. (<Cell mainbuilding33.A1>, <Cell mainbuilding33.B1>)
    8. (<Cell mainbuilding33.A2>, <Cell mainbuilding33.B2>)
    9. (<Cell mainbuilding33.A3>, <Cell mainbuilding33.B3>)

    根据字母获得列号,根据列号返回字母

    需要导入, 这两个函数存在于openpyxl.utils

    1. from openpyxl.utils import get_column_letter, column_index_from_string
    2. # 根据列的数字返回字母
    3. print(get_column_letter(2)) # B
    4. # 根据字母返回列的数字
    5. print(column_index_from_string('D')) # 4

    将数据写入Excel

    工作表相关

    需要导入WorkBook

    1. from openpyxl import Workbook
    2. wb = Workbook()

    这样就新建了一个新的工作表(只是还没被保存)。

    若要指定只写模式,可以指定参数write_only=True。一般默认的可写可读模式就可以了。

    1. print(wb.get_sheet_names()) # 提供一个默认名叫Sheet的表,office2016下新建提供默认Sheet1
    2. # 直接赋值就可以改工作表的名称
    3. sheet.title = 'Sheet1'
    4. # 新建一个工作表,可以指定索引,适当安排其在工作簿中的位置
    5. wb.create_sheet('Data', index=1) # 被安排到第二个工作表,index=0就是第一个位置
    6. # 删除某个工作表
    7. wb.remove(sheet)
    8. del wb[sheet]

    写入单元格

    还可以使用公式哦

    1. # 直接给单元格赋值就行
    2. sheet['A1'] = 'good'
    3. # B9处写入平均值
    4. sheet['B9'] = '=AVERAGE(B2:B8)'

    但是如果是读取的时候需要加上data_only=True这样读到B9返回的就是数字,如果不加这个参数,返回的将是公式本身'=AVERAGE(B2:B8)'

    append函数

    可以一次添加多行数据,从第一行空白行开始(下面都是空白行)写入。

    1. # 添加一行
    2. row = [1 ,2, 3, 4, 5]
    3. sheet.append(row)
    4. # 添加多行
    5. rows = [
    6. ['Number', 'data1', 'data2'],
    7. [2, 40, 30],
    8. [3, 40, 25],
    9. [4, 50, 30],
    10. [5, 30, 10],
    11. [6, 25, 5],
    12. [7, 50, 10],
    13. ]

    由于append函数只能按行写入。如果我们想按列写入呢。append能实现需求么?如果把上面的列表嵌套看作矩阵。只要将矩阵转置就可以了。使用zip()函数可以实现,不过内部的列表变成了元组就是了。都是可迭代对象,不影响。

    1. list(zip(*rows))
    2. # out
    3. [('Number', 2, 3, 4, 5, 6, 7),
    4. ('data1', 40, 40, 50, 30, 25, 50),
    5. ('data2', 30, 25, 30, 10, 5, 10)]

    解释下上面的list(zip(*rows))首先*rows将列表打散,相当于填入了若干个参数,zip从某个列表中提取第1个值组合成一个tuple,再从每个列表中提取第2个值组合成一个tuple,一直到最短列表的最后一个值提取完毕后结束,更长列表的之后的值被舍弃,换句话,最后的元组个数是由原来每个参数(可迭代对象)的最短长度决定的。比如现在随便删掉一个值,最短列表长度为2,data2那一列(竖着看)的值全部被舍弃。

    1. rows = [
    2. ['Number', 'data1', 'data2'],
    3. [2, 40],
    4. [3, 40, 25],
    5. [4, 50, 30],
    6. [5, 30, 10],
    7. [6, 25, 5],
    8. [7, 50, 10],
    9. ]
    10. # out
    11. [('Number', 2, 3, 4, 5, 6, 7), ('data1', 40, 40, 50, 30, 25, 50)]

    最后zip返回的是zip对象,看不到数据的。使用list转换下就好了。使用zip可以方便实现将数据按列写入。

    保存文件

    所有的操作结束后,一定记得保存文件。指定路径和文件名,后缀名为xlsx

    1. wb.save(r'D:example.xlsx')

    设置单元格风格--Style

    先导入需要的类from openpyxl.styles import Font, colors, Alignment

    分别可指定字体相关,颜色,和对齐方式。

    字体

    1. bold_itatic_24_font = Font(name='等线', size=24, italic=True, color=colors.RED, bold=True)
    2. sheet['A1'].font = bold_itatic_24_font

    上面的代码指定了等线24号加粗斜体,字体颜色红色。直接使用cell的font属性,将Font对象赋值给它。

    对齐方式

    也是直接使用cell的属性aligment,这里指定垂直居中和水平居中。除了center,还可以使用right、left等等参数。

    1. # 设置B1中的数据垂直居中和水平居中
    2. sheet['B1'].alignment = Alignment(horizontal='center', vertical='center')

    设置行高和列宽

    有时候数据太长显示不完,就需要拉长拉高单元格。

    1. # 第2行行高
    2. sheet.row_dimensions[2].height = 40
    3. # C列列宽
    4. sheet.column_dimensions['C'].width = 30

    合并和拆分单元格

    所谓合并单元格,即以合并区域的左上角的那个单元格为基准,覆盖其他单元格使之称为一个大的单元格。

    相反,拆分单元格后将这个大单元格的值返回到原来的左上角位置。

    1. # 合并单元格, 往左上角写入数据即可
    2. sheet.merge_cells('B1:G1') # 合并一行中的几个单元格
    3. sheet.merge_cells('A1:C3') # 合并一个矩形区域中的单元格

    合并后只可以往左上角写入数据,也就是区间中:左边的坐标。

    如果这些要合并的单元格都有数据,只会保留左上角的数据,其他则丢弃。换句话说若合并前不是在左上角写入数据,合并后单元格中不会有数据。

    以下是拆分单元格的代码。拆分后,值回到A1位置。

    1. sheet.unmerge_cells('A1:C3')

    这里就拿常用的说,具体的去看openpyxl文档

    阅读原文

    python xlrd,xlwt 读写excel文件

    python 读excel文件,需要xlrd库。下载地址:https://pypi.python.org/pypi/xlrd

    python 写excel文件,需要xlwt库。下载地址:https://pypi.python.org/pypi/xlwt/1.1.2

    下载后修改扩展名为rar, 解压后安装:

    安装成功后就可以引用了。如下代码:

    # -*- coding: utf-8 -*-
    import os
    import xlrd
    import xlwt
    import datetime

    ##################读excel文件##############################

    #打开Excel文件,参数:excelFile:Excel文件路径
    def open_Excel(excelFile):
      excelFile = unicode(excelFile, "utf8")
      if os.path.isfile(excelFile):
        try:
          data = xlrd.open_workbook(excelFile)
          return data
        except Exception,e:
          print str(e)

    '''往EXCEl单元格写内容,每次写一行sheet:页签名称;row:行内容列表;rowIndex:行索引;
    isBold:true:粗字段,false:普通字体'''
    def WriteSheetRow(sheet,rowValueList,rowIndex,isBold):
      i = 0
      style = xlwt.easyxf('font: bold 1') #粗字体

      #style = xlwt.easyxf('font: bold 1, color red;') #红色字体
      for svalue in rowValueList:
        strValue = unicode(str(svalue),'utf-8')
        if isBold:
          sheet.write(rowIndex,i,strValue,style)
        else:
          sheet.write(rowIndex,i,strValue)
        i = i + 1

    #根据索引获取Excel表格中的数据 参数:excelFile:Excel文件路径 ,by_index:表的索引
    def open_Excel_ByIndex(excelFile,sheetIndex):
      data = open_Excel(excelFile)
      table = data.sheets()[sheetIndex]
      nrows = table.nrows #行数
      ncols = table.ncols #列数
      cursor = getSqlCursor()
      for i in xrange(0,nrows):
        headCols = table.row_values(i) #某一行数据 
        for a in headCols:
          print a

    #测试
    open_Excel_ByIndex("D:\test.xlsx",0)

    #根据名称获取Excel表格中的数据 参数:excelFile:Excel文件路径 
    #sheetName:Sheet1名称 
    def open_Excel_BySheetName(excelFile,sheetName):
      sheetName = unicode(sheetName, "utf8")
      data = open_Excel(excelFile)
      table = data.sheet_by_name(sheetName)
      nrows = table.nrows #行数
      ncols = table.ncols #列数
      cursor = getSqlCursor()
      for i in xrange(0,nrows):
        headCols = table.row_values(i) #某一行数据 
        for a in headCols:
          print a
    #测试
    open_Excel_BySheetName("D:\test.xlsx",'sheet1')

    ##################写excel文件##############################
    '''写excel文件''' 
    def save_Excel(strFile):
      excelFile = unicode(strFile, "utf8")
      wbk = xlwt.Workbook()
      sheet = wbk.add_sheet('sheet1')
      headList = ['标题1','标题2','标题3','标题4']
      rowIndex = 0
      WriteSheetRow(sheet,headList,rowIndex,,True)
      for i in xrange(1,11):
        rowIndex = rowIndex + 1
        valueList = []
        for j in xrange(1,5):
          valueList.append(j*i)
        WriteSheetRow(sheet,valueList,rowIndex,False)
      wbk.save(excelFile)

    #测试
    save_Excel("D:\test.xlsx")

    结果如下:

    ##################单元格常用设置##############################

    1、设置超链接

      #设置超链接
      

      font = xlwt.Font() # Create Font
      font.colour_index = 4 # 蓝色字体
      font.underline=True
      style = xlwt.XFStyle()
      style.font = font
      sheet.write(rowIndex,4,xlwt.Formula('HYPERLINK("https://www.baidu.com";"baidu")'),style)

    2、设置单元格背景色

      #设置单元格背景色 
      pattern = xlwt.Pattern()
      pattern.pattern = xlwt.Pattern.SOLID_PATTERN
      pattern.pattern_fore_colour = 5 #黄色
      style = xlwt.XFStyle()
      style.pattern = pattern
      sheet.write(5, 5, 'Cell Contents', style)
      ''' 颜色值
      0 = Black, 1 = White, 2 = Red, 3 = Green, 4 = Blue, 
      5 = Yellow, 6 = Magenta, 7 = Cyan, 16 = Maroon, 
      17 = Dark Green, 18 = Dark Blue, 19 = Dark Yellow ,
      20 = Dark Magenta, 21 = Teal,
      22 = Light Gray, 23 = Dark Gray 
      '''

    3、设置粗字体

      style = xlwt.easyxf('font: bold 1')

      sheet.write(5, 5, 'Cell Contents', style)

    4、设置字体颜色

      style = xlwt.easyxf('font: bold 0, color red;')#红色字体

      sheet.write(5, 5, 'Cell Contents', style)

    5、设置列宽带

      sheet.col(1).width = 3333 # 3333 = 1" (one inch)

    6、设置日期格式

      style = xlwt.XFStyle()
      style.num_format_str = 'YYYY/MM/DD h:mm:ss' 
      sheet.write(5, 5, datetime.datetime.now(), style)

      

      #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

    7、合并行和列

      sheet.write_merge(0, 0, 0, 3, 'First Merge')

      sheet.write_merge(2, 4, 0, 3, 'Second Merge')

      

    8、给单元格增加边框 

      borders = xlwt.Borders() # Create Borders
      borders.left = xlwt.Borders.DASHED

      # May be: NO_LINE, THIN, MEDIUM, DASHED, DOTTED, THICK, DOUBLE, HAIR, MEDIUM_DASHED,   THIN_DASH_DOTTED,              #MEDIUM_DASH_DOTTED, THIN_DASH_DOT_DOTTED, MEDIUM_DASH_DOT_DOTTED,

      #SLANTED_MEDIUM_DASH_DOTTED, or 0x00 through 0x0D.
      borders.right = xlwt.Borders.DASHED
      borders.top = xlwt.Borders.DASHED
      borders.bottom = xlwt.Borders.DASHED
      borders.left_colour = 0x40
      borders.right_colour = 0x40
      borders.top_colour = 0x40
      borders.bottom_colour = 0x40
      style = xlwt.XFStyle() # Create Style
      style.borders = borders # Add Borders to Style

      

    9、设置单元格中内容中位置,居中,局左右等

      alignment = xlwt.Alignment() # Create Alignment
      # May be: HORZ_GENERAL, HORZ_LEFT, HORZ_CENTER, HORZ_RIGHT, 
      #HORZ_FILLED, HORZ_JUSTIFIED, HORZ_CENTER_ACROSS_SEL, HORZ_DISTRIBUTED
      alignment.horz = xlwt.Alignment.HORZ_CENTER   #水平居中
      # May be: VERT_TOP, VERT_CENTER, VERT_BOTTOM, VERT_JUSTIFIED, VERT_DISTRIBUTED
      alignment.vert = xlwt.Alignment.VERT_CENTER    #垂直居中
      style = xlwt.XFStyle() # Create Style
      style.alignment = alignment # Add Alignment to Style
      sheet.write(5, 5, 'Cell Contents', style)

      

  • 相关阅读:
    STL之rb_tree的find函数
    Ruby对象、变量和常量
    Welcome to the Real World
    Git经常使用命令
    Android休眠唤醒机制简介(一)【转】
    DirectFB学习之移植到nuc972平台 标签: DirectFBlinux图形加速驱动【转】
    Buildroot构建指南——工具链【转】
    深入浅出
    android 添加一个按键键值【转】
    Android系统升级那些事儿【转】
  • 原文地址:https://www.cnblogs.com/zhuzhubaoya/p/8718432.html
Copyright © 2011-2022 走看看