python - openpyxl:如何从所有行的特定列中获取数据？

Question

我已经搜索了论坛几次，但没有找到任何涉及在迭代所有行时查找特定列信息的内容。

我有一个包含 8 列和 2200 行的电子表格。

我只需要列 0 5 8 9 (A G J K)

我当前的代码如下，它返回 A-K 中的所有值并删除 None 值。

from openpyxl import Workbook , load_workbook
from itertools import chain

wb = Workbook()

wb = load_workbook("CR_Master_List.xlsx", read_only=True)

ws = wb.active

mylist = []
def iter_rows(ws):
    for row in ws.iter_rows():
        yield [cell.value for cell in row if cell.value is not None]

mylist = list(iter_rows(ws))

print(mylist)

如有任何帮助，我们将不胜感激。

Answer 1

以下内容应该有效。

cols = [0, 5, 8, 9]

def filter_cols(ws):
    for row in ws.iter_rows(max_col=10):
        cells = [cell.value for (idx, cell) in enumerate(row) if (
             idx in cols and cell.value is not None)]
        yield cells