Openpyxl объединить ячейки с одинаковым значением в одном столбце - PullRequest
1 голос
/ 06 мая 2020

Я записал фрейм данных ниже в файл Excel

df = {'name':  ['a', 'b', 'b', 'b', 'c', 'd', 'd', 'e'],
        'type': ['tiger', 'caterpillar', 'butterfly', 'butterfly', 'dolphin', 'tadpole', 'frog', 'dog'],
        'month1_weight': ['33', '0', '0', '0', '59', '3', '0', '0'],
        'month2_weight': ['0', '0', '20', '20', '65', '0', '10', '2'],
        }
dataframe = pd.DataFrame(df)

name      type         month1_weight  month2_weight
a         tiger        33             0
b         caterpillar  0              0
b         butterfly    0              20
b         butterfly    0              20
c         dolphin      59             65
d         tadpole      3              0
d         frog         0              10
e         dog          0              2

enter image description here

Как мне объединить ячейки в столбце имени в Excel и сделать это так:

enter image description here

1 Ответ

1 голос
/ 06 мая 2020

Сначала добавьте многоуровневый индекс в свой фрейм данных, используя set_index(["name", "type"]), затем, используя объект pandas.ExcelWriter, вы можете записать требуемый фрейм данных на лист Excel. Используйте:

from pandas import ExcelWriter

dataframe = pd.DataFrame(df).set_index(["name", "type"])    
with ExcelWriter("excel_file.xlsx") as writer:
    dataframe.to_excel(writer)

После выполнения вышеуказанного кода содержимое вашего excel_file должно выглядеть так:

enter image description here


РЕДАКТИРОВАТЬ (см. Комментарии):

Заменить:

for col_num, value in enumerate(dataframe.columns.values): 
    worksheet_O.write(0, col_num + 1, value, header_format)

на:

for col_num, value in enumerate(dataframe.reset_index().columns): 
    worksheet_O.write(0, col_num, value, header_format) 
...