def regression(data, regtype=0):
xAxis = NumberAxis("x")
xAxis.setAutoRangeIncludesZero(False)
yAxis = NumberAxis("y")
yAxis.setAutoRangeIncludesZero(False)
series = XYSeries("values")
xmax = xmin = None
for (x, y) in data:
series.add(x, y)
if xmax is None:
xmax = xmin = x
else:
xmax = max(xmax, x)
xmin = min(xmin, x)
dataset = XYSeriesCollection()
dataset.addSeries(series)
renderer1 = XYDotRenderer()
plot = XYPlot(dataset, xAxis, yAxis, renderer1)
if regtype == 1:
coefficients = Regression.getPowerRegression(dataset, 0)
curve = PowerFunction2D(coefficients[0], coefficients[1])
regdesc = "Power Regression"
else:
coefficients = Regression.getOLSRegression(dataset, 0)
curve = LineFunction2D(coefficients[0], coefficients[1])
regdesc = "Linear Regression"
regressionData = DatasetUtilities.sampleFunction2D(
curve, xmin, xmax, 100, "Fitted Regression Line")
plot.setDataset(1, regressionData)
renderer2 = XYLineAndShapeRenderer(True, False)
renderer2.setSeriesPaint(0, Color.blue)
plot.setRenderer(1, renderer2)
jfchart = JFreeChart(regdesc, JFreeChart.DEFAULT_TITLE_FONT, plot, True)
chart = Chart(jfchart)
chart.coeffs = coefficients
return chart
def plot(geoms, size=(500,500)):
"""
Plots a set of geometry objects into a xy grid.
*geom* is a `list` of geometry objects to plot. *size* is the resulting
size of the rendered plot.
"""
if not isinstance(geoms, list):
geoms = [geoms]
data = GeometryDataset(geoms)
r = GeometryRenderer()
#r.setFillPolygons(True)
plot = XYPlot(data, data.getDomain(), data.getRange(), r);
chart = JFreeChart(plot)
panel = ChartPanel(chart)
frame = swing.JFrame()
frame.setContentPane(panel)
frame.size = size
frame.visible = True
