使用 openpyxl 加载工作表并忽略包含数据透视表的其他工作表

Question

我有一个 .xlsx 文件，其中包含 2 个工作表。第一个包含常规数据（没什么花哨的），而第二个包含数据透视表。我只需要第一个工作表中的数据，而我想忽略第二个工作表，但是当调用 openpyxl.load_workbook 时，数据透视表会引发错误：TypeError: expected <type 'basestring'>。

错误出现在：openpyxl/reader/excel.py，在行：pivot_caches = parser.pivot_caches。

我尝试使用openpyxl 版本2.6.4 和2.5.1。我正在使用 Python 2.7。

删除第二张工作表后，错误消失了，并且正确读取了第一张工作表中的数据。但是，这些文件是由用户上传的，虽然我不需要数据透视表，但如果可能的话，我想避免强迫用户删除不必要的工作表。

示例代码：

from io import BytesIO

import openpyxl

pivot = '~/Downloads/file_with_pivot_tables.xlsx'

with open(pivot) as fin:
    content = BytesIO(fin.read())
    wb = openpyxl.load_workbook(content)  # this line fails
    ws = wb.get_sheet_by_name('Sheet1')

整个错误跟踪：

  File "/Users/gi/lib/openpyxl/reader/excel.py", line 224, in load_workbook
    pivot_caches = parser.pivot_caches
  File "/Users/gi/lib/openpyxl/packaging/workbook.py", line 125, in pivot_caches
    cache = get_rel(self.archive, self.rels, id=c.id, cls=CacheDefinition)
  File "/Users/gi/lib/openpyxl/packaging/relationship.py", line 162, in get_rel
    obj.deps = get_dependents(archive, rels_path)
  File "/Users/gi/lib/openpyxl/packaging/relationship.py", line 130, in get_dependents
    rels = RelationshipList.from_tree(node)
  File "/Users/gi/lib/openpyxl/descriptors/serialisable.py", line 84, in from_tree
    obj = desc.expected_type.from_tree(el)
  File "/Users/gi/lib/openpyxl/descriptors/serialisable.py", line 100, in from_tree
    return cls(**attrib)
  File "/Users/gi/lib/openpyxl/packaging/relationship.py", line 50, in __init__
    self.Target = Target
  File "/Users/gi/lib/openpyxl/descriptors/base.py", line 44, in __set__
    raise TypeError('expected ' + str(self.expected_type))
TypeError: expected <type 'basestring'>

Answer 1

您可以指定要操作的工作表：

wb = openpyxl.load_workbook('H:\\myfile.xlsx')
ws = wb['sheet1']
ws['E1'] = 'The sky is gray.'
wb.save('H:\\myfile.xlsx')
wb.close()

如果需要先检查，还可以获取所有工作表名称的列表：

print(wb.sheetnames)