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 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 updateChartDataset(self, drawLabels=True):
dataset = XYSeriesCollection()
# self._calculateConfArea()
# dataset.addSeries(self.confArea)
dataset.addSeries(self.markers)
plot = self.chart.getPlot()
rangeAxis = plot.getRangeAxis()
domainAxis = plot.getDomainAxis()
rangeAxis.setRange(self.minY, self.maxY) # change
domainAxis.setRange(self.minX, self.maxX)
# plot.setBackgroundPaint(Color.lightGray);
# plot.setAxisOffset(new RectangleInsets(5.0, 5.0, 5.0, 5.0));
# plot.setDomainGridlinePaint(Color.white);
# plot.setRangeGridlinePaint(Color.white);
markerRenderer = XYLineAndShapeRenderer(False, True)
# print self.markerColor
markerRenderer.setSeriesPaint(0, self.markerColor)
markerRenderer.setSeriesShape(0, Ellipse2D.Double(-3, -3, 6, 6))
# markerRenderer.setToolTipGenerator(FDistToolTipGenerator(self.pointNames))
plot.setRenderer(0, markerRenderer)
plot.setDataset(0, dataset)
dataset = XYSeriesCollection()
if self.drawCI:
dataset = YIntervalSeriesCollection()
CIRenderer = DeviationRenderer(True, False)
# CIRenderer.setOutline(True)
# CIRenderer.setRoundXCoordinates(True)
dataset.addSeries(self.bottom)
dataset.addSeries(self.top)
dataset.addSeries(self.limit)
CIRenderer.setSeriesFillPaint(0, self.balColor)
CIRenderer.setSeriesFillPaint(1, self.neuColor)
CIRenderer.setSeriesFillPaint(2, self.posColor)
CIRenderer.setSeriesPaint(0, self.balColor)
CIRenderer.setSeriesPaint(1, self.neuColor)
CIRenderer.setSeriesPaint(2, self.posColor)
plot.setDataset(1, dataset)
plot.setRenderer(1, CIRenderer)
plot.setDataset(1, dataset)
if drawLabels:
self.drawLabels()
def updateChartDataset(self, drawLabels=True):
dataset = XYSeriesCollection()
#self._calculateConfArea()
#dataset.addSeries(self.confArea)
dataset.addSeries(self.markers)
plot = self.chart.getPlot()
rangeAxis = plot.getRangeAxis()
domainAxis = plot.getDomainAxis()
rangeAxis.setRange(self.minY, self.maxY) #change
domainAxis.setRange(self.minX, self.maxX)
#plot.setBackgroundPaint(Color.lightGray);
#plot.setAxisOffset(new RectangleInsets(5.0, 5.0, 5.0, 5.0));
#plot.setDomainGridlinePaint(Color.white);
#plot.setRangeGridlinePaint(Color.white);
markerRenderer = XYLineAndShapeRenderer(False, True)
#print self.markerColor
markerRenderer.setSeriesPaint(0, self.markerColor)
markerRenderer.setSeriesShape(0, Ellipse2D.Double(-3, -3, 6, 6))
#markerRenderer.setToolTipGenerator(FDistToolTipGenerator(self.pointNames))
plot.setRenderer(0, markerRenderer)
plot.setDataset(0, dataset)
dataset = XYSeriesCollection()
if self.drawCI:
dataset = YIntervalSeriesCollection()
CIRenderer = DeviationRenderer(True, False)
# CIRenderer.setOutline(True)
# CIRenderer.setRoundXCoordinates(True)
dataset.addSeries(self.bottom)
dataset.addSeries(self.top)
dataset.addSeries(self.limit)
CIRenderer.setSeriesFillPaint(0, self.balColor)
CIRenderer.setSeriesFillPaint(1, self.neuColor)
CIRenderer.setSeriesFillPaint(2, self.posColor)
CIRenderer.setSeriesPaint(0, self.balColor)
CIRenderer.setSeriesPaint(1, self.neuColor)
CIRenderer.setSeriesPaint(2, self.posColor)
plot.setDataset(1, dataset)
plot.setRenderer(1, CIRenderer)
plot.setDataset(1, dataset)
if drawLabels:
self.drawLabels()
def scatter(data,x=None,y=None,**kwargs):
''' Creates a scatter plot comparing two elements. At minimum, takes a collection of data.
The second and third arguments, if they exist, are treated as the two elements to
compare. If these arguments do not exist, the first two elements in the list are
compared. If an optional regress=True argument is present, superimposes a linear
regression for each series and prints some related info (R-value etc).
Note that scatter plots can be zoomed with the mouse. Returns the plot object in
case you want to customize the graph in some way. Takes an optional showMissing argument
which determines whether missing values (-9999.0) should be displayed.
Examples: scatter(data), scatter(data,"tmin","tmax",regress=True)
'''
from org.jfree.data.xy import XYSeriesCollection,XYSeries
from org.jfree.data import UnknownKeyException
from org.jfree.chart import ChartFactory,ChartFrame
from org.jfree.chart.plot import PlotOrientation,DatasetRenderingOrder
from org.jfree.chart.renderer.xy import XYLineAndShapeRenderer
from java.awt import Color
regress=kwargs.get('regress',False)
showMissing=kwargs.get('showMissing',False)
# Try to be flexible about element parameters
if x is not None: x = findElement(x).name
if y is not None: y = findElement(y).name
# Create a dataset from the data
collection = XYSeriesCollection()
for ob in data.groupedByObservation().items():
key,values = ob
name = str(key[0])
if x==None:
x = values[0].element.name
try:
xFact = (i for i in values if i.element.name == x).next()
except StopIteration: # missing value
continue
xval = xFact.value
if xval in missingValues and not showMissing: continue
if y==None:
try:
y = values[1].element.name
except IndexError:
raise Exception("Error! Your data request returned only 1 value per observation. "
"Must have 2 values to generate a scatter plot.")
try:
yFact = (i for i in values if i.element.name == y).next()
except StopIteration: # missing value
continue
yval = yFact.value
if yval in missingValues and not showMissing: continue
try:
series = collection.getSeries(name)
except UnknownKeyException:
collection.addSeries(XYSeries(name))
series = collection.getSeries(name)
series.add(float(xval),float(yval))
# Create chart from dataset
chart = ChartFactory.createScatterPlot( "", x, y, collection, PlotOrientation.VERTICAL,
True, True, False );
plot = chart.getPlot()
frame = ChartFrame("Scatter Plot", chart);
frame.pack();
frame.setVisible(True);
# Superimpose regression if desired
if regress:
regressioncollection = XYSeriesCollection()
for series in collection.getSeries():
regression = _getregression(series)
x1 = series.getMinX()
y1 = regression.predict(x1)
x2 = series.getMaxX()
y2 = regression.predict(x2)
regressionseries = XYSeries(series.getKey())
regressionseries.add(float(x1),float(y1))
regressionseries.add(float(x2),float(y2))
regressioncollection.addSeries(regressionseries)
print series.getKey(),":"
print " R: %8.4f" % regression.getR()
print " R-squared: %8.4f" % regression.getRSquare()
print " Significance: %8.4f" % regression.getSignificance()
print
plot.setDataset(1,regressioncollection)
regressionRenderer = XYLineAndShapeRenderer(True,False)
plot.setRenderer(1,regressionRenderer)
plot.setDatasetRenderingOrder(DatasetRenderingOrder.FORWARD);
colors = [0xec0000,0x58b911,0x6886ea,0xedd612,0xa93bb9,0xffb71b,0xe200df,0x1de2b6,0xdc91db,0x383838,0xb09344,0x4ea958,0xd78c9e,0x64008d,0xb0c95b]
mainRenderer = plot.getRenderer(0)
for i in range(collection.getSeriesCount()):
try:
mainRenderer.setSeriesPaint(i,Color(colors[i]))
regressionRenderer.setSeriesPaint(i,Color(colors[i]))
except IndexError: # Finite # of colors in the color array; beyond that let jfreechart pick
break
'''
# Jump through some hoops to ensure regressions are same color as scatters for each series.
# Initially: doesn't work because series are not indexed the same. And I don't see a way
# to get the actual series from the renderer in order to compare names or something.
mainRenderer = plot.getRenderer(0)
print "Renderer is",type(mainRenderer)
index = 0
paint = mainRenderer.lookupSeriesPaint(index)
print "Paint is",type(paint)
while (paint is not None):
print "Setting paint."
regressionRenderer.setSeriesPaint(index,paint)
index += 1
paint = mainRenderer.getSeriesPaint(index)
'''
return plot
