Как добавить запись а excel файл не перезаписывая его?

Question

[Данил Самодуров]

Всем привет. Обрабатываю данные скриптом python:

import os
import pandas as pd

data = pd.read_excel("ОБЩИЙ ЖУРНАЛ.xls")
for i in data:
    if "Unnamed" in i:
        del data[i]
    else:
        continue

for i in data["Протокол, №"]:
    data_protocol = data[data["Протокол, №"] == i]
    graphs_dir = os.path.join(os.getcwd(), "Graphs")
    writer = pd.ExcelWriter(f'{os.path.join(graphs_dir, f"{i}.xlsx")}', engine='xlsxwriter', mode='w')
    try:
        os.mkdir(graphs_dir)
    except OSError:
        pass
    for j in data_protocol["T, °C"]:
        data_temper = data_protocol[data_protocol["T, °C"] == j]
        for k in data_temper["R"]:
            data_R = data_temper[data_temper["R"] == k]
            data_R.to_excel(writer, sheet_name=f'{i}_T={j}°C_R={k}')
    else:
        writer.save()

Скрипт писал давно. По итогу у меня создавалось некоторое количество excel файлов. Теперь нужно снова обработать данные и дописать в ранее созданные файлы. В документации к pandas.ExcelWriter написано, что можно поменять mode='w' на mode='a'. Но после запуска выходит ошибка:
ValueError: Append mode is not supported with xlsxwriter!
Пробовал убирать параметр engine='xlsxwriter', но он по умолчанию и та же сама ошибка вылетает.
У кого-нибудь есть идеи?

Answer 1

[ScriptKiddo]

Pandas умеет добавлять в файлы с движком "openpyxl".
Если openpyxl не установлен - перед запуском скрипта установите его

writer = pd.ExcelWriter(f'{os.path.join(graphs_dir, f"{i}.xlsx")}', engine='openpyxl', mode='a')

UPD. Если нужно добавить данные в уже существующий лист - вот сниппет со Stackoverflow
https://stackoverflow.com/a/38075046

Code

def append_df_to_excel(filename, df, sheet_name='Sheet1', startrow=None,
                       truncate_sheet=False, 
                       **to_excel_kwargs):
    """
    Append a DataFrame [df] to existing Excel file [filename]
    into [sheet_name] Sheet.
    If [filename] doesn't exist, then this function will create it.

    Parameters:
      filename : File path or existing ExcelWriter
                 (Example: '/path/to/file.xlsx')
      df : dataframe to save to workbook
      sheet_name : Name of sheet which will contain DataFrame.
                   (default: 'Sheet1')
      startrow : upper left cell row to dump data frame.
                 Per default (startrow=None) calculate the last row
                 in the existing DF and write to the next row...
      truncate_sheet : truncate (remove and recreate) [sheet_name]
                       before writing DataFrame to Excel file
      to_excel_kwargs : arguments which will be passed to `DataFrame.to_excel()`
                        [can be dictionary]

    Returns: None
    """
    from openpyxl import load_workbook

    import pandas as pd

    # ignore [engine] parameter if it was passed
    if 'engine' in to_excel_kwargs:
        to_excel_kwargs.pop('engine')

    writer = pd.ExcelWriter(filename, engine='openpyxl')

    # Python 2.x: define [FileNotFoundError] exception if it doesn't exist 
    try:
        FileNotFoundError
    except NameError:
        FileNotFoundError = IOError


    try:
        # try to open an existing workbook
        writer.book = load_workbook(filename)

        # get the last row in the existing Excel sheet
        # if it was not specified explicitly
        if startrow is None and sheet_name in writer.book.sheetnames:
            startrow = writer.book[sheet_name].max_row

        # truncate sheet
        if truncate_sheet and sheet_name in writer.book.sheetnames:
            # index of [sheet_name] sheet
            idx = writer.book.sheetnames.index(sheet_name)
            # remove [sheet_name]
            writer.book.remove(writer.book.worksheets[idx])
            # create an empty sheet [sheet_name] using old index
            writer.book.create_sheet(sheet_name, idx)

        # copy existing sheets
        writer.sheets = {ws.title:ws for ws in writer.book.worksheets}
    except FileNotFoundError:
        # file does not exist yet, we will create it
        pass

    if startrow is None:
        startrow = 0

    # write out the new sheet
    df.to_excel(writer, sheet_name, startrow=startrow, **to_excel_kwargs)

    # save the workbook
    writer.save()

Comments

[Данил Самодуров]

Спасибо!