Передать переменную в add_series в xlsxwriter - PullRequest
Новая
       7

Передать переменную в add_series в xlsxwriter

0 голосов
/ 09 января 2019

Я пытаюсь использовать цикл for для добавления ряда для диаграммы в файл 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
    })

Заранее спасибо!

Ответы [ 2 ]

0 голосов
/ 10 января 2019

Лучший способ установить диапазон диаграммы 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],
    })
0 голосов
/ 09 января 2019

Я нашел ответ на 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
    })
...