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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
|
#!/usr/bin/env python
# Eclipse SUMO, Simulation of Urban MObility; see https://eclipse.org/sumo
# Copyright (C) 2008-2020 German Aerospace Center (DLR) and others.
# This program and the accompanying materials are made available under the
# terms of the Eclipse Public License 2.0 which is available at
# https://www.eclipse.org/legal/epl-2.0/
# This Source Code may also be made available under the following Secondary
# Licenses when the conditions for such availability set forth in the Eclipse
# Public License 2.0 are satisfied: GNU General Public License, version 2
# or later which is available at
# https://www.gnu.org/licenses/old-licenses/gpl-2.0-standalone.html
# SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-or-later
# @file plot_csv_bars.py
# @author Daniel Krajzewicz
# @date 2014-01-27
"""
This script plots name / value pairs from a given .csv file (';'-separated).
The values are plotted as bars.
matplotlib (http://matplotlib.org/) has to be installed for this purpose
"""
from __future__ import absolute_import
from __future__ import print_function
import os
import sys
sys.path.append(os.path.dirname(os.path.dirname(os.path.realpath(__file__))))
import sumolib # noqa
from sumolib.visualization import helpers # noqa
import matplotlib.pyplot as plt # noqa
def main(args=None):
"""The main function; parses options and plots"""
# ---------- build and read options ----------
from optparse import OptionParser
optParser = OptionParser()
optParser.add_option("-i", "--input", dest="input", metavar="FILE",
help="Defines the csv file to use as input")
optParser.add_option("-c", "--column", dest="column",
type="int", default=1, help="Selects the column to read values from")
optParser.add_option("-r", "--revert", dest="revert", action="store_true",
default=False, help="Reverts the order of read values")
optParser.add_option("-w", "--width", dest="width",
type="float", default=.8, help="Defines the width of the bars")
optParser.add_option("--space", dest="space",
type="float", default=.2, help="Defines the space between the bars")
optParser.add_option("--norm", dest="norm",
type="float", default=1., help="Divides the read numbers by this value before plotting them")
optParser.add_option("--show-values", dest="showValues", action="store_true",
default=False, help="Shows the values")
optParser.add_option("--values-offset", dest="valuesOffset",
type="float", default=1., help="Position offset for values")
optParser.add_option("--vertical", dest="vertical", action="store_true",
default=False, help="vertical bars are used")
optParser.add_option("-v", "--verbose", dest="verbose", action="store_true",
default=False, help="If set, the script says what it's doing")
# standard plot options
helpers.addInteractionOptions(optParser)
helpers.addPlotOptions(optParser)
# parse
options, remaining_args = optParser.parse_args(args=args)
if options.input is None:
print("Error: at least one csv file must be given")
sys.exit(1)
fd = open(options.input)
labels = []
vlabels = []
vals = []
total = 0
xs = []
ts = []
s = options.width + options.space
t = options.width / 2. + options.space / 2.
x = options.space / 2.
for line in fd:
v = line.strip().split(";")
if len(v) < 2:
continue
labels.append(v[0].replace("\\n", "\n"))
value = float(v[options.column]) / options.norm
vals.append(value)
vlabels.append(str(value) + "%")
total += value
xs.append(x)
ts.append(t)
x = x + s
t = t + s
if options.revert:
labels.reverse()
vals.reverse()
vlabels.reverse()
colors = []
for i, e in enumerate(labels):
colors.append(helpers.getColor(options, i, len(labels)))
fig, ax = helpers.openFigure(options)
if not options.vertical:
rects = plt.barh(xs, vals, height=options.width)
for i, rect in enumerate(rects):
if options.showValues:
width = rect.get_width()
ax.text(width + options.valuesOffset, rect.get_y() +
rect.get_height() / 2., vlabels[i], va='center', ha='left')
rect.set_color(colors[i])
rect.set_edgecolor('k')
plt.ylim(0, x)
plt.yticks(ts, labels)
else:
rects = plt.bar(xs, vals, width=options.width)
for i, rect in enumerate(rects):
if options.showValues:
height = rect.get_height()
ax.text(rect.get_x() + rect.get_width() / 2., height +
options.valuesOffset, vlabels[i], ha='center', va='bottom')
rect.set_color(colors[i])
rect.set_edgecolor('k')
plt.xlim(0, x)
plt.xticks(ts, labels)
helpers.closeFigure(fig, ax, options, False)
if __name__ == "__main__":
sys.exit(main(sys.argv))
|
