def constructImage(self):
bar1 = Bar()
bar1.xValues = range(5)
bar1.yValues = [2, 4, 6, 8, 10]
bar1.color = "red"
bar1.label = "Red Cluster"
bar2 = Bar()
bar2.xValues = range(5)
bar2.yValues = [3, 12, 4, 8, 14]
bar2.color = "blue"
bar2.label = "Blue Cluster"
bar3 = Bar()
bar3.xValues = range(5)
bar3.yValues = [1, 6, 9, 13, 20]
bar3.color = "green"
bar3.label = "Green Cluster"
clusteredBars = ClusteredBars()
clusteredBars.add(bar1)
clusteredBars.add(bar2)
clusteredBars.add(bar3)
clusteredBars.spacing = 0.5
clusteredBars.xTickLabels = ["A", "B", "C", "D", "E"]
plot = Plot()
plot.add(clusteredBars)
plot.hasLegend()
plot.save(self.imageName)
def getBinBars(rows,
binKey,
quantityKey,
xLabelFormattingFunction,
xLabel,
yLabel,
xLabelRotation=0):
bins = []
countsPerBin = []
for row in rows:
bins.append(int(row[binKey]))
countsPerBin.append(int(row[quantityKey]))
bar = Bar()
bar.xValues = bins
bar.yValues = countsPerBin
if xLabelFormattingFunction is not None:
bar.xTickLabelPoints = range(len(bins))
bar.xTickLabels = map(xLabelFormattingFunction, bar.xTickLabelPoints)
if xLabelRotation > 0:
bar.setXTickLabelProperties(rotation=str(xLabelRotation))
barPlot = Plot()
barPlot.add(bar)
barPlot.setXLabel(xLabel)
barPlot.setYLabel(yLabel)
# barPlot.setXLimits(0, None)
return barPlot
def constructImage(self):
plot = Plot()
# Uneven error bars
line = Line()
line.xValues = [6, 10, 4, 0, 8, 2, 12]
line.yValues = [50, 90, 30, 10, 70, 20, 110]
line.yMins = [y - 30 for y in line.yValues]
line.yMaxes = [y + 50 for y in line.yValues]
line.label = "Asymmetric Errors"
line.color = "red"
# Even error bars
line2 = Line()
line2.xValues = [1, 5, 3, 9, 7, 11]
line2.yValues = [100, 120, 110, 140, 130, 150]
line2.color = "blue"
line2.label = "Symmetric Errors"
line2.yErrors = [5, 25, 15, 45, 35, 55]
plot.add(line)
plot.add(line2)
plot.xLabel = "X Label"
plot.yLabel = "Y Label"
plot.hasLegend()
plot.xLimits = (-1, 13)
plot.save(self.imageName)
def constructImage(self):
scatter = Scatter()
scatter.label="Hooray dots with vlines!"
scatter.xValues = [
7, 6, 3, 5, 7, 6, 1, 1, 6, 5, 8, 6, 7, 8, 0, 2, 9, 3, 9,
5, 4, 5, 0, 0, 2, 3, 1, 4, 1, 3, 3, 8, 1, 5, 2, 6, 0, 3,
5, 1, 4, 9, 5, 1, 9, 9, 9, 7, 6, 5, 8, 6, 0, 2, 6, 9, 2,
5, 6, 9, 7, 8, 7, 6, 5, 9, 9, 2, 4, 9, 0, 1, 1, 1, 6, 4,
5, 8, 9, 1, 2, 1, 4, 5, 9, 7, 4, 9, 2, 9, 2, 5, 2, 2, 0,
2, 1, 9, 3, 6]
scatter.yValues = [
2, 5, 9, 5, 9, 3, 6, 1, 6, 0, 0, 6, 2, 5, 3, 9, 2, 7, 6,
2, 3, 1, 9, 9, 5, 2, 9, 0, 2, 3, 0, 2, 5, 5, 8, 4, 1, 9,
8, 6, 1, 6, 9, 2, 4, 9, 2, 8, 1, 1, 2, 1, 0, 6, 3, 4, 2,
5, 6, 8, 6, 9, 0, 6, 8, 6, 8, 1, 6, 2, 2, 3, 6, 2, 2, 2,
0, 2, 4, 6, 8, 5, 1, 4, 2, 3, 5, 3, 1, 0, 6, 0, 1, 6, 8,
9, 3, 9, 3, 7]
vline1 = VLine()
vline1.xValues = [2,8]
vline1.color = 'CornflowerBlue'
vline2 = VLine()
vline2.xValues = [1,9]
vline2.color = 'GoldenRod'
plot = Plot()
plot.hasLegend()
plot.add(scatter)
plot.add(vline1)
plot.add(vline2)
plot.save(self.imageName)
def constructImage(self):
bar1 = Bar()
bar1.xValues = range(5)
bar1.yValues = [1, 2, 1, 2, 3]
bar1.color = "red"
bar1.label = "Red Cluster"
bar2 = Bar()
bar2.xValues = range(5)
bar2.yValues = [2, 2, 3, 3, 4]
bar2.color = "blue"
bar2.label = "Blue Cluster"
bar3 = Bar()
bar3.xValues = range(5)
bar3.yValues = [3, 5, 4, 5, 3]
bar3.color = "green"
bar3.label = "Green Cluster"
stackedBars = StackedBars()
stackedBars.add(bar1)
stackedBars.add(bar2)
stackedBars.add(bar3)
stackedBars.xTickLabels = ["A", "B", "C", "D", "E"]
plot = Plot()
plot.add(stackedBars)
plot.setYLimits(0, 15)
plot.hasLegend()
plot.save(self.imageName)
def constructImage(self):
layout = PlotLayout()
plotBase10 = Plot()
plotBase10.loglog = True
lineBase10 = Line()
lineBase10.marker = 'x'
lineBase10.xValues = [1, 10, 100, 1000, 10000]
lineBase10.yValues = [1, 25, 140, 1024, 10342]
plotBase10.add(lineBase10)
plotBase2 = Plot()
plotBase2.logx = True
plotBase2.logbase = 2
lineBase2 = Line()
lineBase2.marker = 'x'
lineBase2.xValues = [1, 2, 4, 8, 16, 32, 64]
lineBase2.yValues = [1, 2, 3, 4, 5, 6, 7]
plotBase2.add(lineBase2)
layout.addPlot(plotBase10)
layout.addPlot(plotBase2)
layout.width = 2
layout.save(self.imageName)
def constructImage(self):
scatter = Scatter()
scatter.label = "Hooray dots with vlines!"
scatter.xValues = [
7, 6, 3, 5, 7, 6, 1, 1, 6, 5, 8, 6, 7, 8, 0, 2, 9, 3, 9, 5, 4, 5,
0, 0, 2, 3, 1, 4, 1, 3, 3, 8, 1, 5, 2, 6, 0, 3, 5, 1, 4, 9, 5, 1,
9, 9, 9, 7, 6, 5, 8, 6, 0, 2, 6, 9, 2, 5, 6, 9, 7, 8, 7, 6, 5, 9,
9, 2, 4, 9, 0, 1, 1, 1, 6, 4, 5, 8, 9, 1, 2, 1, 4, 5, 9, 7, 4, 9,
2, 9, 2, 5, 2, 2, 0, 2, 1, 9, 3, 6
]
scatter.yValues = [
2, 5, 9, 5, 9, 3, 6, 1, 6, 0, 0, 6, 2, 5, 3, 9, 2, 7, 6, 2, 3, 1,
9, 9, 5, 2, 9, 0, 2, 3, 0, 2, 5, 5, 8, 4, 1, 9, 8, 6, 1, 6, 9, 2,
4, 9, 2, 8, 1, 1, 2, 1, 0, 6, 3, 4, 2, 5, 6, 8, 6, 9, 0, 6, 8, 6,
8, 1, 6, 2, 2, 3, 6, 2, 2, 2, 0, 2, 4, 6, 8, 5, 1, 4, 2, 3, 5, 3,
1, 0, 6, 0, 1, 6, 8, 9, 3, 9, 3, 7
]
vline1 = VLine()
vline1.xValues = [2, 8]
vline1.color = 'CornflowerBlue'
vline2 = VLine()
vline2.xValues = [1, 9]
vline2.color = 'GoldenRod'
plot = Plot()
plot.hasLegend()
plot.add(scatter)
plot.add(vline1)
plot.add(vline2)
plot.save(self.imageName)
def constructImage(self):
bar1 = Bar()
bar1.xValues = range(5)
bar1.yValues = [1, 2, 1, 2, 3]
bar1.color = "red"
bar1.label = "Red Cluster"
bar2 = Bar()
bar2.xValues = range(5)
bar2.yValues = [2, 2, 3, 3, 4]
bar2.color = "blue"
bar2.label = "Blue Cluster"
bar3 = Bar()
bar3.xValues = range(5)
bar3.yValues = [3, 5, 4, 5, 3]
bar3.color = "green"
bar3.label = "Green Cluster"
stackedBars = StackedBars()
stackedBars.add(bar1)
stackedBars.add(bar2)
stackedBars.add(bar3)
stackedBars.xTickLabels = ["A", "B", "C", "D", "E"]
plot = Plot()
plot.add(stackedBars)
plot.yLimits = (0, 15)
plot.hasLegend()
plot.save(self.imageName)
def main():
output_graph = sys.argv[1]
plot = Plot()
plot.hasLegend(location='lower right')
plot.xLabel = 'Per-client throughput (Mbps)' # Change this
plot.yLabel = 'CDF'
plot.xLimits = (0, 50)
plot.yLimits = (0, 1)
plot.legendLabelSize = FONT_SIZE
plot.xTickLabelSize = FONT_SIZE - 2
plot.yTickLabelSize = FONT_SIZE - 2
plot.axesLabelSize = FONT_SIZE
for csv_file in sys.argv[2:]:
cdf_table = _make_cdf(csv_file)
line = Line()
line.xValues = [x for (x, _) in cdf_table]
line.yValues = [y for (_, y) in cdf_table]
line.color = colors.pop(0)
line.lineStyle = line_styles.pop(0)
# Extract the filename
line.label = capitalize( csv_file.split('/')[-2].replace('.csv', '') )
plot.add(line)
plot.save(output_graph)
def constructImage(self):
plot = Plot()
# Uneven error bars
line = Line()
line.xValues = [6,10,4,0,8,2,12]
line.yValues = [50,90,30,10,70,20,110]
line.yMins = [y - 30 for y in line.yValues]
line.yMaxes = [y + 50 for y in line.yValues]
line.label = "Asymmetric Errors"
line.color = "red"
# Even error bars
line2 = Line()
line2.xValues = [1,5,3,9,7,11]
line2.yValues = [100, 120, 110, 140, 130, 150]
line2.color = "blue"
line2.label = "Symmetric Errors"
line2.yErrors = [5,25,15,45,35,55]
plot.add(line)
plot.add(line2)
plot.xLabel = "X Label"
plot.yLabel = "Y Label"
plot.hasLegend()
plot.xLimits = (-1, 13)
plot.save(self.imageName)
def histogram_plot(experiment_log_dir, plot_spec_string, output_filename,
has_legend, x_limit, verbose):
queries = [
plot_utils.plot_spec_string_to_query(plot_spec_string, 0, "HIST")]
plot_data = metaprogram_utils.process_queries(
queries, experiment_log_dir, verbose)
if "plot_points" not in plot_data:
warnings.warn("No data to plot!")
return
histogram_data = plot_data["plot_points"][0]
cumulative_histogram = {}
layout = PlotLayout()
layout.dpi = 250
for stat_name in histogram_data:
plot = Plot()
plot.setTitle(stat_name)
if has_legend:
plot.hasLegend(labelSize=8)
if x_limit is not None:
plot.setXLimits(0, x_limit)
style_plot(plot, stat_name)
for key, points in sorted(histogram_data[stat_name].items()):
for size, count in itertools.izip(points["bin"], points["count"]):
if size not in cumulative_histogram:
cumulative_histogram[size] = 0
cumulative_histogram[size] += count
line = Line()
line.stepFunction("pre")
line.label = str(key)
line.xValues = points["bin"]
line.yValues = points["count"]
plot.add(line)
layout.addPlot(plot)
cumulative_plot = Plot()
if x_limit is not None:
cumulative_plot.setXLimits(0, x_limit)
cumulative_plot.setTitle("Cumulative Histogram for " + stat_name)
style_plot(cumulative_plot, stat_name)
line = Line()
line.stepFunction("pre")
line.xValues = sorted(cumulative_histogram.keys())
line.yValues = [cumulative_histogram[key] for key in line.xValues]
cumulative_plot.add(line)
layout.addPlot(cumulative_plot)
layout.save(output_filename)
def constructImage(self):
layout = PlotLayout()
plotBase10 = Plot()
plotBase10.loglog = True
lineBase10 = Line()
lineBase10.marker = 'x'
lineBase10.xValues = [1, 10, 100, 1000, 10000]
lineBase10.yValues = [1, 25, 140, 1024, 10342]
plotBase10.add(lineBase10)
plotBase2 = Plot()
plotBase2.logx = True
plotBase2.logbase = 2
lineBase2 = Line()
lineBase2.marker = 'x'
lineBase2.xValues = [1, 2, 4, 8, 16, 32, 64]
lineBase2.yValues = [1, 2, 3, 4, 5, 6, 7]
plotBase2.add(lineBase2)
layout.addPlot(plotBase10)
layout.addPlot(plotBase2)
layout.width = 2
layout.save(self.imageName)
def constructImage(self):
line = Line()
line.xValues = [2, 1, 3, 4, 0]
line.yValues = [2, 1, 3, 4, 0]
plot = Plot()
plot.add(line)
plot.save(self.imageName)
def constructImage(self):
line = Line()
line.yValues = [25, 40, 30, 23, 10, 50]
line.xValues = range(len(line.yValues))
plot = Plot()
plot.add(line)
plot.xLabel = "X Label"
plot.yLabel = "Y Label"
plot.yLimits = (0, 60)
plot.save(self.imageName)
def constructImage(self):
plot = Plot()
bar = Bar()
bar.xValues = range(5)
bar.yValues = [2, 8, 4, 6, 5]
plot.add(bar)
plot.xLabel = "Widget ID"
plot.yLabel = "# Widgets Sold"
plot.save(self.imageName)
def constructImage(self):
plot = Plot()
plot.projection = 'polar'
r = arange(0,1,0.001)
theta = 2*2*pi*r
line = Line()
line.xValues = theta
line.yValues = r
plot.add(line)
plot.save(self.imageName)
def constructImage(self):
bar = Bar()
bar.xValues = range(5)
bar.yValues = [2, 4, 6, 8, 10]
# Valid values include all marker types, /, //, \, \\
bar.hatch = r"\\"
bar.color="red"
bar.edgeColor="black"
plot = Plot()
plot.add(bar)
plot.save(self.imageName)
def constructImage(self):
plot = Plot()
line = Line()
line.yValues = [25, 40, 30, 23, 10, 50]
line.xValues = range(len(line.yValues))
plot.add(line)
plot.setXLabel("X Label")
plot.setYLabel("Y Label")
plot.setYLimits(0, 60)
plot.save(self.imageName)
def constructImage(self):
plot = Plot()
line = Line()
line.yValues = [25, 40, 30, 23, 10, 50]
line.xValues = range(len(line.yValues))
line.xTickLabels = ["X 1", "X 2", "X 3", "X 4", "X 5"]
line.yTickLabels = ["Y Ten", "Y Twenty", "Y Thirty", "Y Forty",
"Y Fifty", "Y Sixty"]
line.yTickLabelPoints = [10, 20, 30, 40, 50, 60]
# You can set tick label properties with a dictionary ...
line.xTickLabelProperties = {
"color" : "blue",
"weight" : "bold",
"rotation" : 45
}
line.yTickLabelProperties = {
"style" : "italic",
"alpha" : 0.5,
"color" : "red"
}
# (clearing for demonstrative purposes)
line.xTickLabelProperties.clear()
line.yTickLabelProperties.clear()
# You can also set by direct elementwise access
line.xTickLabelProperties["color"] = "blue"
line.xTickLabelProperties["weight"] = "bold"
line.xTickLabelProperties["rotation"] = "45"
line.yTickLabelProperties["style"] = "italic"
line.yTickLabelProperties["alpha"] = 0.5
line.yTickLabelProperties["color"] = "red"
plot.add(line)
plot.title = "Craaazy Title"
plot.setTitleProperties(
style="italic", weight="bold", rotation="5",
color="orange")
plot.xLabel = "X Label"
plot.yLabel = "Y Label"
plot.yLimits = (0, 60)
plot.tight = True
plot.save(self.imageName)
def generatePlot(stepType):
line = Line()
line.xValues = xVals
line.yValues = yVals
line.marker = 'o'
line.stepFunction(stepType)
plot = Plot()
plot.add(line)
plot.title = r'"%s" Steps' % (stepType)
plot.xLimits = (0, 6)
plot.yLimits = (0, 6)
return plot
def constructImage(self):
plot = Plot()
line = Line()
line.yValues = [25, 40, 30, 23, 10, 50]
line.xValues = range(len(line.yValues))
line.xTickLabels = ["X 1", "X 2", "X 3", "X 4", "X 5"]
line.yTickLabels = ["Y Ten", "Y Twenty", "Y Thirty", "Y Forty",
"Y Fifty", "Y Sixty"]
line.yTickLabelPoints = [10, 20, 30, 40, 50, 60]
# You can set tick label properties with a dictionary ...
line.xTickLabelProperties = {
"color" : "blue",
"weight" : "bold",
"rotation" : 45
}
line.yTickLabelProperties = {
"style" : "italic",
"alpha" : 0.5,
"color" : "red"
}
# (clearing for demonstrative purposes)
line.xTickLabelProperties.clear()
line.yTickLabelProperties.clear()
# You can also set by direct elementwise access
line.xTickLabelProperties["color"] = "blue"
line.xTickLabelProperties["weight"] = "bold"
line.xTickLabelProperties["rotation"] = "45"
line.yTickLabelProperties["style"] = "italic"
line.yTickLabelProperties["alpha"] = 0.5
line.yTickLabelProperties["color"] = "red"
plot.add(line)
plot.setTitle("Craaazy Title")
plot.setTitleProperties(style="italic", weight="bold", rotation="5",
color="orange")
plot.setXLabel("X Label")
plot.setYLabel("Y Label")
plot.setYLimits(0, 60)
plot.setPlotParameters(bottom=.15, left=0.15)
plot.save(self.imageName)
def generatePlot(stepType):
line = Line()
line.xValues = xVals
line.yValues = yVals
line.marker = 'o'
line.stepFunction(stepType)
plot = Plot()
plot.add(line)
plot.setTitle(r'"%s" Steps' % (stepType))
plot.setXLimits(0, 6)
plot.setYLimits(0, 6)
return plot
def constructImage(self):
line = Line()
line.xValues = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
line.yValues = [0, 1, 4, 9, 16, 25, 36, 49, 64, 81, 100]
plot = Plot()
plot.useLatexLabels()
plot.xLabel = r"$x$"
plot.yLabel = r"$f(x) = x^2$"
plot.title = (r"LaTeX is Number $\sum_{n=1}^{\infty}"
r"\frac{-e^{i\pi}}{2^n}$")
plot.add(line)
plot.tight = True
plot.axesLabelSize = 18
plot.save(self.imageName)
def constructImage(self):
cluster = ClusteredBars()
colors = ['red','green','blue','CornflowerBlue','LightSalmon']
yVals = [[
[1, 3, 2, 5, 4],
[2, 2, 2, 2, 2],
[1, 3, 2, 4, 3],
[0, 4, 0, 4, 0],
[5, 5, 5, 5, 5]
],[
[2, 2, 2, 2, 2],
[2, 2, 2, 2, 2],
[2, 2, 2, 2, 2],
[2, 2, 2, 2, 2],
[2, 2, 2, 2, 2]
],
[
[1, 3, 1, 3, 1],
[1, 3, 1, 3, 1],
[1, 3, 1, 3, 1],
[1, 3, 1, 3, 1],
[1, 3, 1, 3, 1],
]]
for i in xrange(3):
stack = StackedBars()
for j in xrange(5):
bar = Bar()
bar.xValues = range(5)
bar.yValues = yVals[i][j]
bar.color = colors[j]
bar.label = "Subject %d" % (j+1,)
stack.add(bar)
cluster.add(stack)
cluster.spacing = 0.5
cluster.xTickLabels = ["1", "2", "3", "4", "5"]
plot = Plot()
plot.add(cluster)
plot.hasLegend()
plot.setXLabel('Nested Cars')
plot.setYLabel('Party (lampshades)')
plot.save(self.imageName)
def constructImage(self):
plot = Plot()
line = Line()
line.yValues = [25, 40, 30, 23, 10, 50]
line.xValues = range(len(line.yValues))
plot.add(line)
plot.xLabel = "X Label"
plot.yLabel = "Y Label"
plot.yLimits = (0, 60)
plot.grid.color = "#ff0000"
plot.grid.style = "dotted"
plot.grid.visible = True
plot.save(self.imageName)
def constructImage(self):
plot = Plot()
line = Line()
line.xValues = xrange(100)
line.xTickLabels = ["Whoa this label is really long why is this label so long"]
line.xTickLabelPoints = [42]
line.xTickLabelProperties["rotation"] = 45
line.yValues = [math.sin(x) for x in xrange(100)]
line.yTickLabels = ["Look at this value. Pretty sweet value right?"]
line.yTickLabelPoints = [0.3]
plot.add(line)
plot.setXLabel("Value")
plot.setYLabel("sin(Value)")
plot.save(self.imageName)
def constructImage(self):
plot = Plot()
line = Line()
line.yValues = [25, 40, 30, 23, 10, 50]
line.xValues = range(len(line.yValues))
plot.add(line)
plot.xLabel = "X Label"
plot.yLabel = "Y Label"
plot.yLimits = (0, 60)
plot.grid.color = "#ff0000"
plot.grid.style = "dotted"
plot.grid.visible = True
plot.save(self.imageName)
def constructImage(self):
plot = Plot()
for i in xrange(6):
line = Line()
line.xValues = xrange(5)
line.yValues = [(i + 1) * x for x in line.xValues]
line.label = "Line %d" % (i + 1)
plot.add(line)
plot.addLineColor("red")
plot.addLineColor("blue")
plot.addLineColor("green")
plot.addMarker("")
plot.addMarker("x")
plot.hasLegend(columns=2)
plot.save(self.imageName)
def constructImage(self):
line = Line()
line.xValues = [0, 1, 2, 3, 4, 5,
6, 7, 8, 9, 10]
line.yValues = [0, 1, 4, 9, 16, 25,
36, 49, 64, 81, 100]
plot = Plot()
plot.useLatexLabels()
plot.xLabel = r"$x$"
plot.yLabel = r"$f(x) = x^2$"
plot.title = (
r"LaTeX is Number $\sum_{n=1}^{\infty}"
r"\frac{-e^{i\pi}}{2^n}$")
plot.add(line)
plot.tight = True
plot.axesLabelSize = 18
plot.save(self.imageName)
def constructImage(self):
plot = Plot()
line = Line()
line.yValues = [25, 40, 30, 23, 10, 50]
line.xValues = range(len(line.yValues))
plot.add(line)
plot.xLabel = "X Label"
plot.yLabel = "Y Label"
plot.yLimits = (0, 60)
plot.xTickLabelSize = 24
plot.yTickLabelSize = 36
plot.axesLabelSize = 18
plot.tight = True
plot.save(self.imageName)
def constructImage(self):
plot = Plot()
line = Line()
line.yValues = [25, 40, 30, 23, 10, 50]
line.xValues = range(len(line.yValues))
plot.add(line)
plot.setXLabel("X Label")
plot.setYLabel("Y Label")
plot.setYLimits(0, 60)
plot.setXTickLabelSize(24)
plot.setYTickLabelSize(36)
plot.setAxesLabelSize(18)
plot.setPlotParameters(bottom=0.14)
plot.save(self.imageName)
def constructImage(self):
plot = Plot()
line = Line()
line.xValues = xrange(100)
line.xTickLabels = [
"Whoa this label is really long why is this label so long"
]
line.xTickLabelPoints = [42]
line.xTickLabelProperties["rotation"] = 45
line.yValues = [math.sin(x) for x in xrange(100)]
line.yTickLabels = ["Look at this value. Pretty sweet value right?"]
line.yTickLabelPoints = [0.3]
plot.add(line)
plot.setXLabel("Value")
plot.setYLabel("sin(Value)")
plot.save(self.imageName)
def testExactSize(self):
plot = Plot()
line = Line()
plot.setDimensions(3, 4)
line.xValues = range(5)
line.yValues = range(5)
plot.add(line)
plot.save(self.imageName)
im = Image.open(self.imageName)
self.assertEqual(im.size, (300,400))
plot.setDimensions(3,4,dpi=250)
plot.save(self.imageName)
im = Image.open(self.imageName)
self.assertEqual(im.size, (750,1000))
def testExactSize(self):
plot = Plot()
line = Line()
plot.setDimensions(3, 4)
line.xValues = range(5)
line.yValues = range(5)
plot.add(line)
plot.save(self.imageName)
im = Image.open(self.imageName)
self.assertEqual(im.size, (300, 400))
plot.setDimensions(3, 4, dpi=250)
plot.save(self.imageName)
im = Image.open(self.imageName)
self.assertEqual(im.size, (750, 1000))
def constructImage(self):
line = Line()
line.xValues = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
line.yValues = [0, 1, 4, 9, 16, 25, 36, 49, 64, 81, 100]
plot = Plot()
plot.useLatexLabels()
plot.setXLabel(r"$x$")
plot.setYLabel(r"$f(x) = x^2$")
plot.setTitle(r"LaTeX is Number $\sum_{n=1}^{\infty}\frac{-e^{i\pi}}{2^n}$")
plot.add(line)
layout = PlotLayout()
layout.useLatexLabels()
layout.addPlot(plot)
layout.setAxesLabelSize(18)
layout.setPlotParameters(top=0.84)
layout.save(self.imageName)
def constructImage(self):
plot = Plot()
for i in xrange(24):
line = Line()
line.xValues = xrange(5)
line.yValues = [(i + 1) * x for x in line.xValues]
line.label = "Line %d" % (i + 1)
plot.add(line)
plot.addLineColor("red")
plot.addLineColor("blue")
plot.addLineStyle("-")
plot.addLineStyle("dashed")
plot.addLineStyle("dotted")
plot.addMarker('none')
plot.addMarker('x')
plot.addMarker('o')
plot.hasLegend(columns=2)
plot.setLegendLabelSize(8)
plot.save(self.imageName)
def constructImage(self):
line = Line()
line.xValues = numpy.arange(0, 150, 0.01)
line.yValues = numpy.cos(.02 * numpy.pi * line.xValues)
plot = Plot()
plot.add(line)
plot.xLimits = (0, 150)
plot.yLimits = (-1, 1)
plot.xLabel = "X"
plot.yLabel = "cos(X)"
splitPlots = plot.split(2)
layout = PlotLayout()
layout.width = 2
layout.addPlot(plot, grouping="unsplit")
for s in splitPlots:
layout.addPlot(s, grouping="splits")
layout.save(self.imageName)
def constructImage(self):
plot = Plot()
for i in xrange(24):
line = Line()
line.xValues = xrange(5)
line.yValues = [(i+1) * x for x in line.xValues]
line.label = "Line %d" % (i + 1)
plot.add(line)
plot.addLineColor("red")
plot.addLineColor("blue")
plot.addLineStyle("-")
plot.addLineStyle("dashed")
plot.addLineStyle("dotted")
plot.addMarker('none')
plot.addMarker('x')
plot.addMarker('o')
plot.hasLegend(columns=2)
plot.setLegendLabelSize(8)
plot.save(self.imageName)
def constructImage(self):
line = Line()
line.xValues = numpy.arange(0, 150, 0.01)
line.yValues = numpy.cos(.02 * numpy.pi * line.xValues)
plot = Plot()
plot.add(line)
plot.xLimits = (0, 150)
plot.yLimits = (-1, 1)
plot.xLabel = "X"
plot.yLabel = "cos(X)"
splitPlots = plot.split(2)
layout = PlotLayout()
layout.width = 2
layout.addPlot(plot, grouping="unsplit")
for s in splitPlots:
layout.addPlot(s, grouping="splits")
layout.save(self.imageName)
def constructImage(self):
lines = []
for i in xrange(3):
line = Line()
line.xValues = xrange(5)
line.yValues = [(i+1 / 2.0) * pow(x, i+1) for x in line.xValues]
line.label = "Line %d" % (i + 1)
lines.append(line)
plot = Plot()
plot.add(lines[0])
inset = Plot()
inset.add(lines[1])
inset.hideTickLabels()
inset.title = ("Inset in Yo Inset\nSo You Can Inset\nWhile You Inset")
insideInset = Plot()
insideInset.hideTickLabels()
insideInset.add(lines[2])
inset.addInset(insideInset, width=0.4, height=0.3,
location="upper left")
plot.addInset(inset, width=0.4, height=0.4, location="lower right")
plot.save(self.imageName)
def constructImage(self):
line = Line()
line.xValues = range(5)
line.yValues = [2, 3, 5, 7, 9]
line.label = "A Line"
linePlot1 = Plot()
linePlot1.title = "Small Legend"
linePlot1.add(line)
linePlot1.hasLegend()
linePlot1.legendLabelSize = 10
linePlot2 = Plot()
linePlot2.title = "Large Legend"
linePlot2.add(line)
linePlot2.hasLegend()
linePlot2.legendLabelSize = 30
linePlot3 = Plot()
linePlot3.title = "Inherited from Layout"
linePlot3.add(line)
linePlot3.hasLegend()
layout = PlotLayout()
layout.width = 2
layout.addPlot(linePlot1)
layout.addPlot(linePlot2)
layout.addPlot(linePlot3)
layout.legendLabelSize = 15
layout.save(self.imageName)
def constructImage(self):
line = Line()
line.xValues = range(5)
line.yValues = [2,3,5,7,9]
line.label = "A Line"
linePlot1 = Plot()
linePlot1.title = "Small Legend"
linePlot1.add(line)
linePlot1.hasLegend()
linePlot1.legendLabelSize = 10
linePlot2 = Plot()
linePlot2.title = "Large Legend"
linePlot2.add(line)
linePlot2.hasLegend()
linePlot2.legendLabelSize = 30
linePlot3 = Plot()
linePlot3.title = "Inherited from Layout"
linePlot3.add(line)
linePlot3.hasLegend()
layout = PlotLayout()
layout.width = 2
layout.addPlot(linePlot1)
layout.addPlot(linePlot2)
layout.addPlot(linePlot3)
layout.legendLabelSize = 15
layout.save(self.imageName)
def constructImage(self):
line = Line()
line.xValues = range(5)
line.yValues = [2,3,5,7,9]
line.label = "A Line"
linePlot1 = Plot()
linePlot1.setTitle("Small Legend")
linePlot1.add(line)
linePlot1.hasLegend()
linePlot1.setLegendLabelSize(10)
linePlot2 = Plot()
linePlot2.setTitle("Large Legend")
linePlot2.add(line)
linePlot2.hasLegend()
linePlot2.setLegendLabelSize(30)
linePlot3 = Plot()
linePlot3.setTitle("Inherited from Layout")
linePlot3.add(line)
linePlot3.hasLegend()
layout = PlotLayout()
layout.setWidth(2)
layout.addPlot(linePlot1)
layout.addPlot(linePlot2)
layout.addPlot(linePlot3)
layout.setLegendLabelSize(15)
layout.setPlotParameters(left=0.03, bottom=0.03, right=0.98, top=0.94)
layout.save(self.imageName)
def constructImage(self):
lines = []
for i in xrange(3):
line = Line()
line.xValues = xrange(5)
line.yValues = [(i+1 / 2.0) * pow(x, i+1) for x in line.xValues]
line.label = "Line %d" % (i + 1)
lines.append(line)
plot = Plot()
plot.add(lines[0])
inset = Plot()
inset.add(lines[1])
inset.hideTickLabels()
inset.setTitle("Inset in Yo Inset\nSo You Can Inset\nWhile You Inset")
insideInset = Plot()
insideInset.hideTickLabels()
insideInset.add(lines[2])
inset.addInset(insideInset, width=0.4, height=0.3,
location="upper left")
plot.addInset(inset, width=0.4, height=0.4, location="lower right")
plot.save(self.imageName)
def constructImage(self):
cluster = ClusteredBars()
colors = ['red', 'green', 'blue', 'CornflowerBlue', 'LightSalmon']
yVals = [[[1, 3, 2, 5, 4], [2, 2, 2, 2, 2], [1, 3, 2, 4, 3],
[0, 4, 0, 4, 0], [5, 5, 5, 5, 5]],
[[2, 2, 2, 2, 2], [2, 2, 2, 2, 2], [2, 2, 2, 2, 2],
[2, 2, 2, 2, 2], [2, 2, 2, 2, 2]],
[
[1, 3, 1, 3, 1],
[1, 3, 1, 3, 1],
[1, 3, 1, 3, 1],
[1, 3, 1, 3, 1],
[1, 3, 1, 3, 1],
]]
for i in xrange(3):
stack = StackedBars()
for j in xrange(5):
bar = Bar()
bar.xValues = range(5)
bar.yValues = yVals[i][j]
bar.color = colors[j]
bar.label = "Subject %d" % (j + 1, )
stack.add(bar)
cluster.add(stack)
cluster.spacing = 0.5
cluster.xTickLabels = ["1", "2", "3", "4", "5"]
plot = Plot()
plot.add(cluster)
plot.hasLegend()
plot.setXLabel('Nested Cars')
plot.setYLabel('Party (lampshades)')
plot.save(self.imageName)
def constructImage(self):
line1 = Line()
line2 = Line()
line3 = Line()
line1.xValues = range(0, 10)
line1.yValues = [2, 5, 2, 3, 2, 2, 1, 0, 1, 0]
line2.xValues = range(0, 10)
line2.yValues = [3, 1, 2, 3, 2, 1, 5, 3, 1, 7]
line3.xValues = range(0, 10)
line3.yValues = [2, 1, 3, 1, 3, 4, 1, 4, 5, 0]
stack = StackedLines()
stack.addLine(line1, "red")
stack.addLine(line2, "green")
stack.addLine(line3, "blue")
plot = Plot()
plot.xLimits = (0, 9)
plot.yLimits = (0, 7)
plot.add(stack)
plot.save(self.imageName)
def constructImage(self):
line1 = Line()
line1.xValues = range(7)
line1.yValues = [1, 2, 4, 8, 16, 32, 64]
line1.label = "First Plot"
line1.lineStyle = "-"
line1.color = "red"
line2 = Line()
line2.xValues = range(7)
line2.yValues = [100, 90, 80, 70, 60, 50, 40]
line2.label = "Second Plot"
line2.lineStyle = "--"
line2.color = "blue"
plot = Plot()
plot.add(line1)
plot.add(line2)
plot.xLabel = "Shared X Axis"
plot.yLabel = "First Plot's Y Axis"
plot.setTwinX("Second Plot's Y Axis", 1)
plot.hasLegend()
plot.save(self.imageName)
def constructImage(self):
scatter = Scatter()
scatter.label="Hooray dots!"
scatter.xValues = [
7, 6, 3, 5, 7, 6, 1, 1, 6, 5, 8, 6, 7, 8, 0, 2, 9, 3, 9,
5, 4, 5, 0, 0, 2, 3, 1, 4, 1, 3, 3, 8, 1, 5, 2, 6, 0, 3,
5, 1, 4, 9, 5, 1, 9, 9, 9, 7, 6, 5, 8, 6, 0, 2, 6, 9, 2,
5, 6, 9, 7, 8, 7, 6, 5, 9, 9, 2, 4, 9, 0, 1, 1, 1, 6, 4,
5, 8, 9, 1, 2, 1, 4, 5, 9, 7, 4, 9, 2, 9, 2, 5, 2, 2, 0,
2, 1, 9, 3, 6]
scatter.yValues = [
2, 5, 9, 5, 9, 3, 6, 1, 6, 0, 0, 6, 2, 5, 3, 9, 2, 7, 6,
2, 3, 1, 9, 9, 5, 2, 9, 0, 2, 3, 0, 2, 5, 5, 8, 4, 1, 9,
8, 6, 1, 6, 9, 2, 4, 9, 2, 8, 1, 1, 2, 1, 0, 6, 3, 4, 2,
5, 6, 8, 6, 9, 0, 6, 8, 6, 8, 1, 6, 2, 2, 3, 6, 2, 2, 2,
0, 2, 4, 6, 8, 5, 1, 4, 2, 3, 5, 3, 1, 0, 6, 0, 1, 6, 8,
9, 3, 9, 3, 7]
plot = Plot()
plot.hasLegend()
plot.add(scatter)
plot.save(self.imageName)
def constructImage(self):
line = Line()
line.xValues = range(5)
line.yValues = [2, 4, 6, 8, 10]
linePlot = Plot()
linePlot.add(line)
linePlot.xLabel = "X Data"
linePlot.yLabel = "Y Data"
linePlot.title = "Data as Line"
bar = Bar()
bar.xValues = range(5)
bar.yValues = [2, 4, 6, 8, 10]
barPlot = Plot()
barPlot.add(bar)
barPlot.xLabel = "X Data"
barPlot.yLabel = "Y Data"
barPlot.title = "Data as Bars"
scatter = Scatter()
scatter.xValues = range(5)
scatter.yValues = [2, 4, 6, 8, 10]
scatterPlot = Plot()
scatterPlot.add(scatter)
scatterPlot.xLabel = "X Data"
scatterPlot.yLabel = "Y Data"
scatterPlot.title = "Data as Points"
layout = WeightedPlotLayout()
# Plots in the same grouping are placed together on the same line
layout.addPlot(linePlot, grouping="topRow", weight=2)
layout.addPlot(barPlot, grouping="topRow")
# Plots without a grouping are arranged as follows:
# * While you can make a row of N plots, where N is the size of the plot
# grouping with the largest size, do so.
# * If you can't make a row of N plots, make the plots stretch across a
# single row.
layout.addPlot(scatterPlot)
layout.save(self.imageName)
def constructImage(self):
line = Line()
line.xValues = numpy.arange(0.0, 5.0, 0.01)
line.yValues = numpy.cos(2 * numpy.pi * line.xValues)
maxLabel = Label(2, 1, "Maximum!")
maxLabel.textOffset = (0.5, 0.5)
maxLabel.hasArrow()
minLabel = Label(1.5, -1, "Minimum!")
minLabel.textPosition = (1, -2)
minLabel.hasArrow()
randomLabel = Label(2, -1.7, "A Point!")
randomLabel.textOffset = (0, 0.2)
randomLabel.marker = 'o'
styledLabel = Label(1.25,
1.2,
"A FancyPoint!",
bbox={
'edgecolor': 'red',
'facecolor': 'white',
'ls': 'dashed',
'lw': '2'
})
styledLabel.textOffset = (0, 0.2)
styledLabel.marker = 'o'
plot = Plot()
plot.add(line)
plot.add(minLabel)
plot.add(maxLabel)
plot.add(randomLabel)
plot.add(styledLabel)
plot.yLimits = (-3, 3)
plot.xLabel = "X"
plot.yLabel = "cos(x)"
plot.save("label.png")
plot.save(self.imageName)
def constructImage(self):
line = Line()
line.xValues = range(5)
line.yValues = [2, 4, 6, 8, 10]
linePlot = Plot()
linePlot.add(line)
linePlot.xLabel = "X Data"
linePlot.yLabel = "Y Data"
linePlot.title = "Data as Line"
bar = Bar()
bar.xValues = range(5)
bar.yValues = [2, 4, 6, 8, 10]
barPlot = Plot()
barPlot.add(bar)
barPlot.xLabel = "X Data"
barPlot.yLabel = "Y Data"
barPlot.title = "Data as Bars"
scatter = Scatter()
scatter.xValues = range(5)
scatter.yValues = [2, 4, 6, 8, 10]
scatterPlot = Plot()
scatterPlot.add(scatter)
scatterPlot.xLabel = "X Data"
scatterPlot.yLabel = "Y Data"
scatterPlot.title = "Data as Points"
layout = PlotLayout()
layout.addPlot(linePlot, grouping="topRow")
layout.addPlot(barPlot, grouping="topRow")
layout.addPlot(scatterPlot)
layout.save(self.imageName)
def histogram_plot(experiment_log_dir, plot_spec_string, output_filename,
has_legend, x_limit, verbose):
queries = [
plot_utils.plot_spec_string_to_query(plot_spec_string, 0, "HIST")
]
plot_data = metaprogram_utils.process_queries(queries, experiment_log_dir,
verbose)
if "plot_points" not in plot_data:
warnings.warn("No data to plot!")
return
histogram_data = plot_data["plot_points"][0]
cumulative_histogram = {}
layout = PlotLayout()
layout.dpi = 250
for stat_name in histogram_data:
plot = Plot()
plot.setTitle(stat_name)
if has_legend:
plot.hasLegend(labelSize=8)
if x_limit is not None:
plot.setXLimits(0, x_limit)
style_plot(plot, stat_name)
for key, points in sorted(histogram_data[stat_name].items()):
for size, count in itertools.izip(points["bin"], points["count"]):
if size not in cumulative_histogram:
cumulative_histogram[size] = 0
cumulative_histogram[size] += count
line = Line()
line.stepFunction("pre")
line.label = str(key)
line.xValues = points["bin"]
line.yValues = points["count"]
plot.add(line)
layout.addPlot(plot)
cumulative_plot = Plot()
if x_limit is not None:
cumulative_plot.setXLimits(0, x_limit)
cumulative_plot.setTitle("Cumulative Histogram for " + stat_name)
style_plot(cumulative_plot, stat_name)
line = Line()
line.stepFunction("pre")
line.xValues = sorted(cumulative_histogram.keys())
line.yValues = [cumulative_histogram[key] for key in line.xValues]
cumulative_plot.add(line)
layout.addPlot(cumulative_plot)
layout.save(output_filename)
