python - OpenPyXL - 如何根据某些条件从 Excel 文件中删除行?
问题描述
我想从excel文件中删除行,知道这些值。我使用openpyxl:
key_values_list
是带有数字的列表(所有都存在于 excel 文件中的列上)
wb = load_workbook(src)
sheet = wb['Sheet 1']
for i in range(2, sheet.max_row + 1):
if sheet.cell(row=i, column=1).value in key_values_list:
sheet.delete_rows(i, 1)
wb.save(src)
上面的代码不会删除所有对应的行
解决方案
在 for 循环中删除直接元素总是会出现问题。考虑此代码用于具有 12 行的工作表及其各自的行值:
for i in range(1, sh.max_row + 1):
print(sh.cell(row=i, column=1).value)
# 1 .. 12
现在看看当你开始删除东西时会发生什么:
for i in range(1, sh.max_row + 1):
if sh.cell(row=i, column=1).value in [5,6,7]:
sh.delete_rows(i, 1)
print(f'i = {i}\tcell value (i, 1) is {sh.cell(row=i, column=1).value}')
# i = 1 cell value (i, 1) is 1
# i = 2 cell value (i, 1) is 2
# i = 3 cell value (i, 1) is 3
# i = 4 cell value (i, 1) is 4
# i = 5 cell value (i, 1) is 5
# i = 6 cell value (i, 1) is 7
# i = 7 cell value (i, 1) is 9
# i = 8 cell value (i, 1) is 10
# i = 9 cell value (i, 1) is 11
# i = 10 cell value (i, 1) is 12
# i = 11 cell value (i, 1) is None
# i = 12 cell value (i, 1) is None
您可以看到,在 期间i in [5, 6, 7]
,行的移动从第 6 行开始,因为第 5 行已经被删除,使原来的第 6 行成为新的第 5 行,原来的第 7 行成为新的第 6 行……等等。所以在下一次迭代中i = 6
,单元格实际上是从原始数据中引用第 7 行的值。您有效地跳过了第 6 行的迭代。
最简单的答案是使用while
循环,而不是for
:
i = 1
while i <= sh.max_row:
print(f'i = {i}\tcell value (i, 1) is {sh.cell(row=i, column=1).value}')
if sh.cell(row=i, column=1).value in [5,6,7]:
sh.delete_rows(i, 1)
# Note the absence of incremental. Because we deleted a row, we want to stay on the same row because new data will show in the next iteration.
else:
i += 1
# Because the check failed, we can safely increment to the next row.
# i = 1 cell value (i, 1) is 1
# i = 2 cell value (i, 1) is 2
# i = 3 cell value (i, 1) is 3
# i = 4 cell value (i, 1) is 4
# i = 5 cell value (i, 1) is 5 # deleted
# i = 5 cell value (i, 1) is 6 # deleted
# i = 5 cell value (i, 1) is 7 # deleted
# i = 5 cell value (i, 1) is 8
# i = 6 cell value (i, 1) is 9
# i = 7 cell value (i, 1) is 10
# i = 8 cell value (i, 1) is 11
# i = 9 cell value (i, 1) is 12
# verify the data has been deleted
for i in range(1, sh.max_row +1):
print(sh.cell(row=i, column=1).value)
# 1
# 2
# 3
# 4
# 8
# 9
# 10
# 11
# 12
您现在可以看到虽然i
没有达到 12,但每一行都被迭代,因为i=5
已经处理了三次。
如果出于某种原因您必须使用for
循环进行迭代,您可能需要考虑一些替代方法,例如遍历副本或向后执行
推荐阅读
- postgresql - 复合类型中的可选属性 (Postgresql)
- sql - 如何更改 Azure SQL 在线时间
- python - 在 python 上生成特定图的问题
- node.js - 安装 puppeteer 会引发错误,并显示“无法安装脚本 'node install.js'”
- python-3.x - 从 Django DeleteView 返回到 Django ListView 分页列表中的右侧页面
- java - 导入与当前类同名的类
- python - 使用打印功能的缩进逻辑
- python - 使用 Selenium-Python 检查页面中是否存在文本的最佳方法
- algorithm - 测试一组中的哪些圆相交
- ml - 在 Poly/ML 编程中计算没有重复的元素列表