Я пытаюсь использовать цикл для добавления ряда для диаграммы в файл xlsx, но получаю странную ошибку.
Вот мой код
В этой части показано, что я отсортировал свой DF и переиндексировал его для простоты интерпретации индекса DF для индекса Excel для ссылки на ячейку Excel.
# Sort dataframe and reindex
pivDF = pivDF.sort_values(by=['Vg'])
pivDF = pivDF.reset_index(drop=True) # reindex dropping previous index instead of adding it as a new column
# Add the following as a sheet in a excel file:
# data, chart
## Create workbook
workbook = xlsxwriter.Workbook(dirPath + file_name + '.xlsx') # Creates workbook object
## Add a worksheet to hold the data.
writer = pd.ExcelWriter(dirPath+"//"+file_name+".xlsx", engine = 'xlsxwriter')
pivDF.to_excel(writer,sheet_name = 'Data')
writer.save()
## Add plot
### Create a new line chart.
chartsheet = workbook.add_chartsheet() # creates chartsheet objec
chart = workbook.add_chart({'type': 'line'}) # creates chart object
Следующее добавляет серию, и именно здесь я сталкиваюсь с проблемой.
### Configure the chart
#### Configure series.
##### Get unique Vg
Vglist = pivDF.Vg.unique()
C1 = 'D'
C2 = 'F'
for Vg in Vglist:
# Get min and max row excel index for current Vg
indexList = list(pivDF.loc[(pivDF.Vg == Vglist[0])].index.values) # gets list of index at Vg value
r1 = min(indexList) + 2
r2 = max(indexList) + 2
chart.add_series({
'name': 'Vg ' + Vg + 'V',
'categories': '=Sheet1!$'+ C1 +'$'+ r1 +':$'+ C1 +'$'+ r2, # Vd data
'values': '=Sheet1!$'+ C2 +'$'+ r1 +':$'+ C2 +'$'+ r2, # Id normalized data
})
Заранее спасибо!
Лучший способ установить диапазон диаграммы XlsxWriter программным способом - это избежать синтаксиса диапазона стиля Sheet1!D1:D12
и использовать альтернативный синтаксис list
показанный в документах.
Что-то вроде:
C1 = 3
C2 = 5
for Vg in Vglist:
indexList = list(pivDF.loc[(pivDF.Vg == Vglist[0])].index.values)
r1 = min(indexList) + 1
r2 = max(indexList) + 1
chart.add_series({
'name': 'Vg %sV' % Vg,
'categories': ['Sheet1', r1, C1, r2, C1],
'values': ['Sheet1', r1, C2, r2, C2],
})
Я нашел ответ на https://riptutorial.com/python/example/17392/create-excel-charts-with-xlsxwriter
Код должен выглядеть так:
C1 = 'D'
C2 = 'F'
for Vg in Vglist:
# Get min and max row excel index for current Vg
indexList = list(pivDF.loc[(pivDF.Vg == Vglist[0])].index.values) # gets list of index at Vg value
r1 = min(indexList) + 2
r2 = max(indexList) + 2
chart.add_series({
'name': 'Vg %sV' % Vg,
'categories': '=Sheet1!$%s$%s:$%s$%s' % (C1, r1, C1, r2), # Vd data
'values': '=Sheet1!$%s$%s:$%s$%s' % (C2, r1, C2, r2), # Id normalized data
})