【发布时间】:2015-12-24 16:10:41
【问题描述】:
我正在使用我编写的以下类来尝试动态创建一个包含多个工作表的 Excel 文件,其中每个工作表中都有一个打印的数据框和一个柱形图。
与代码的交互(见下文)应该在您启动工作簿的地方起作用:
test = Workbook('Test Workbook')
然后,您可以添加任意数量的图表:
test.add_chart(df, 'Df Title', 1)
test.add_chart(df2, 'Df2 Title', 1)
然后你生成工作簿:
test.produce()
输入数据帧有标题。第一列是文本类别,后续列(数量不定)是小数形式的数据,以百分比形式绘制。
问题:该代码运行良好,并且似乎将所有工作表与图表分开生成,但某些图表显示为“未引用”,这意味着当我单击柱形图中的条形时,它不突出显示源数据。用代码生成的一些图表,DO引用得当,所以我不确定问题出在哪里,也没有明显的趋势。
import xlsxwriter
import pandas as pd
class Workbook:
def __init__(self, workbook_name):
self.workbook_name = workbook_name
self.workbook = xlsxwriter.Workbook(str(self.workbook_name) + '.xlsx')
self.letters = ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P']
def produce(self):
self.workbook.close()
print 'Created ' + str(self.workbook_name) + '.xlsx'
def print_frame(self, worksheet, dataframe, df_width, start_data_index):
col_as_lists = []
col_names = list(dataframe.columns.values)
# loops through columns in df and converts to list
for n in range(0, df_width):
col_n = dataframe[col_names[n]].tolist()
# checks to see if column has numbers, if so -> convert to float!
if n < start_data_index:
col_n.insert(0, col_names[n])
elif self.is_number(col_n[0]):
convert = col_n[0:]
convert = [float(x) for x in convert]
convert.insert(0, col_names[n])
col_n = convert
else:
col_n.insert(0, col_names[n])
col_as_lists.append(col_n)
# Prints each list into the worksheet.
worksheet.write_column(self.letters[n] + '1', col_as_lists[n])
#Formats numerical data as percentage
percentformat = self.workbook.add_format({'num_format': '0%'})
worksheet.set_column(self.letters[start_data_index] + ':' + self.letters[df_width], None, percentformat)
def add_chart(self, dataframe, tab_name, start_data_index):
df_width = len(dataframe.columns)
worksheet = self.workbook.add_worksheet(tab_name)
self.print_frame(worksheet, dataframe, df_width, start_data_index)
chart = self.workbook.add_chart({'type': 'column'})
df_length = (len(dataframe.index))
for n in range(start_data_index, df_width):
chart.add_series({
'name': '=' + tab_name +'!$' + self.letters[n] + '$1',
'categories': '=' + tab_name +'!$' + self.letters[start_data_index - 1] + '$2:$'+ self.letters[start_data_index - 1] + '$' + str(df_length + 1),
'values': '=' + tab_name +'!$' + self.letters[n] + '$2:$'+ self.letters[n] + '$' + str(df_length + 1),
'fill': {'color': '#FFB11E'},
'data_labels': {'value': True, 'center': True}
})
chart.set_title({'name': tab_name})
chart.set_x_axis({'major_gridlines': {'visible': False}})
chart.set_y_axis({'major_gridlines': {'visible': False}, 'max': .70})
worksheet.insert_chart(self.letters[df_width + 2] + '2', chart)
return
def is_number(self, s):
""" Function used to help with detecting and converting floats
from string to number data types."""
try:
float(s)
return True
except ValueError:
return False
【问题讨论】:
-
看似“未引用”的图表:这些图表是否真的正确,除了当您单击它们时没有突出显示它们的源数据?您的某些工作表名称中是否包含空格,而有些则没有?
-
哇,是的,这似乎是个问题!当我从工作表名称中删除空格时,所有图表都正确显示!关于您的问题,是的,除了参考问题和显示为小数而不是百分比的数据标签之外,图表显示正确。
标签: python excel xlsxwriter