python - 使用 pd.xlsxwriter 根据该行中的单个单元格有条件地格式化整行

Question

我有以下代码:

import pandas as pd
from datetime import datetime
 writer = pd.ExcelWriter(
        f"~/{datetime.today().strftime('%Y.%m.%d')} - Report.xlsx",
        engine="xlsxwriter",
        date_format="dd/mm/yyyy",
        datetime_format="dd/mm/yyyy",
    )
worksheet = writer.sheets["Sheet1"]
comma_format = workbook.add_format(
        {"num_format": "#,###;[Red](#,###);-", "bold": True}
    )
worksheet.conditional_format(
        "$A$2:$J$1000",
        {
            "type": "cell",
            "criteria": "containing",
            "value": '"Total"',
            "format": comma_format,
        },
    )

我想，每当一行中的任何单元格中包含单词“Total”时，使用 comma_format 格式化整个单元格。

然而，上面的脚本只格式化包含“总计”的单元格，而不是整行。

关于如何解决这个问题有什么想法吗？

非常感谢，

迈克

Answer 1

您可以像这样使用 DataFrame 操作来获取包含单词 Total 的行索引。 rows_with_total = df[df.applymap(lambda x: True if isinstance(x, str) and 'total' in x.lower() else False).any(1)].index.to_list().

applymap 迭代 DataFrame 中的所有值并将传入的函数应用于每个值。在这种情况下，如果值是一个字符串并且包含单词 total，则返回 True，否则返回 False。

传入 1 的 any 方法检查一行中的任何值是否为 True 并返回 Series 的 booleans(如果行中的任何列为 True，则为 True，否则为 False)。将该部分包装在 df['that part'] 中，根据 Series boolean 值过滤返回的行。

然后 .index.to_list() 将过滤后的行的索引作为 list 返回。

有了这个列表，您就可以按照@drew_wood 的建议进行操作，并循环遍历行，像这样将格式应用于每一行。

for row in rows_with_total:
    worksheet.set_row(row, None, comma_format)

完整代码

rows_with_total = df[df.applymap(lambda x: True if isinstance(x, str) and 'total' in x.lower() else False).any(1)].index.to_list()
for row in rows_with_total:
    worksheet.set_row(row, None, comma_format)