INNER CODE UNIT · Python
that
abhijithneilabraham/tableQA · tableqa/chart.py:84
Function that returns a pie chart.
# Arguments
columns: `list` object, column names.
answers: `list` object, corresponding answer values.
"""
for i, colname in enumerate(columns):
answer = answers[i]
if len(answer) == 1:
import warnings
warnings.warn('Cannot plot chart for a single value.')
break
plt.figure(figsize=self.figsize)
answer = pd.DataFrame(answer, columns=[colname])
grouped_col = answer[colname].value_counts()
ax = grouped_col.plot(kind='pie', startangle=0, autopct='%.2f', title=colname, fontsize=10, labels=None)
ax.yaxis.set_label_coords(-0.15, 0.5)