Python基础之文件I/O-excel

文件I/O-excel

读写数据

1. 安装 openpyxl

       pip install openpyxl
   

2. 打开文件

       from openpyxl import load_workbook
   
       from openpyxl.writer.excel import ExcelWriter
   
       try:
           workbook_ = load_workbook(u"data.xlsx")
           # 获得表单名字
           sheetnames = workbook_.get_sheet_names()
   
           sheet = workbook_.get_sheet_by_name(sheetnames[0])
       except Exception as ex:
           print(ex)
   
   

3. 获取单元格

       # 获取某个单元格的值，观察excel发现也是先字母再数字的顺序，即先列再行
       b4 = sheet['A4']  #A行4列
   
       # 除了用下标的方式获得，还可以用cell函数, 换成数字，这个表示B4
       b4_too = sheet.cell(row=4, column=2)
       print(b4_too.value)
   
   
       for i in range(2):
           for j in range(5):
               print(sheet.cell(row=i+1, column=j+1).value,end=' ')
           print()
   

4. 获得最大行和最大列

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

5. 获取行和列

       sheet.rows为生成器, 里面是每一行的数据，每一行又由一个tuple包裹。
       sheet.columns类似，不过里面是每个tuple是每一列的单元格。
   

       # 因为按行，所以返回A1, B1, C1这样的顺序
       for row in sheet.rows:
           for cell in row:
               print(cell.value)
   
       # A1, A2, A3这样的顺序
       for column in sheet.columns:
           for cell in column:
               print(cell.value)
   

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

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

6. 切片获取

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

       for row_cell in sheet['A1':'B3']:
           for cell in row_cell:
               print(cell)
   

写入数据

1. 完整案例

       from openpyxl import Workbook
   
       from openpyxl import load_workbook
   
       from openpyxl.writer.excel import ExcelWriter
   
       try:
   
           #     新建工作表
           wb=Workbook()
           # 新建一个工作表，可以指定索引，适当安排其在工作簿中的位置
           sheet=wb.create_sheet('Data', index=1)  # 被安排到第二个工作表，index=0就是第一个位置
   
           # 删除某个工作表
           # wb.remove(wb)
           # del wb[sheet]
           # row = [1, 2, 3, 4, 5]
           # sheet.append(row)
           rows = [
               ['Number', 'data1', 'data2'],
               [2, 40],
               [3, 40, 25],
               [4, 50, 30],
               [5, 30, 10],
               [6, 25, 5],
               [7, 50, 10111],
           ]
           for row in rows:
               sheet.append(row)
           wb.save(r'新歌检索失败1477881109469.xlsx')
       except Exception as ex:
           print(ex)