1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
import matplotlib.pyplot as plt
import sys
filename = sys.argv[1]
title = sys.argv[2]
label = []
value = []
color = []
with open(filename, "r") as f:
for line in f:
tokens = line.split()
c = tokens[-1]
problem_data = " ".join(tokens[0:-1])
l, v = problem_data.split(":")
if int(v) > 0:
label.append(l + "(" + v + ")")
value.append(v)
color.append(c)
wedges, texts, autotexts = plt.pie(
value,
colors=color,
autopct=lambda p: "{:.1f}%".format(round(p)) if p > 0 else "",
pctdistance=1.2,
wedgeprops={"edgecolor": "black", "linewidth": 1, "antialiased": False},
)
plt.legend(wedges, label, loc="center left", bbox_to_anchor=(1.1, 0, 0.5, 1))
# Add title
plt.title(title, pad=20)
if len(sys.argv) <= 3:
plt.legend(
label,
bbox_to_anchor=(1.1, 0.5),
loc="center right",
fontsize=10,
bbox_transform=plt.gcf().transFigure,
shadow=True,
)
else:
plt.legend(
label,
loc="lower center",
bbox_to_anchor=(0.5, -0.1),
fontsize=10,
bbox_transform=plt.gcf().transFigure,
shadow=True,
)
# Show plot
# plt.tight_layout()
plt.savefig(
filename.replace(".txt", ".png"), dpi=600, transparent=True, bbox_inches="tight"
)
|
