In [1]:

%useLatestDescriptors
%use lets-plot
//%use krangl

In [2]:

LetsPlot.getInfo()

Out[2]:

Lets-Plot Kotlin API v.3.2.0. Frontend: Notebook with dynamically loaded JS. Lets-Plot JS v.2.3.0.

In [3]:

%use krangl

In [4]:

val mpg_df = DataFrame.readCSV("https://raw.githubusercontent.com/JetBrains/lets-plot-kotlin/master/docs/examples/data/mpg.csv")
mpg_df.head()

Out[4]:

	manufacturer	model	displ	year	cyl	trans	drv	cty	hwy	fl	class
1	audi	a4	1.8	1999	4	auto(l5)	f	18	29	p	compact
2	audi	a4	1.8	1999	4	manual(m5)	f	21	29	p	compact
3	audi	a4	2.0	2008	4	manual(m6)	f	20	31	p	compact
4	audi	a4	2.0	2008	4	auto(av)	f	21	30	p	compact
5	audi	a4	2.8	1999	6	auto(l5)	f	16	26	p	compact

Shape: 5 x 12.

In [5]:

val mpg = mpg_df.toMap()

In [6]:

// Default legend
val p = letsPlot(mpg) {x="displ"; y="hwy"; color="manufacturer"} + geomPoint(size=5.0) + ggsize(600,250)
p

Out[6]:

In [7]:

// Layout the legend in two columns
p + scaleColorDiscrete(guide=guideLegend(ncol=2)) + ggtitle("Two columns legend")

Out[7]:

In [8]:

// Same and fill by rows
p + scaleColorDiscrete(guide=guideLegend(ncol=2, byRow=true)) +
ggtitle("Two columns legend filled by rows")

Out[8]:

In [9]:

// Adjust legend and axis
//  - five rows legend
//  - draw legend below plot
//  - remove axis
p + scaleColorDiscrete(guide=guideLegend(nrow=5)) +
    theme(axisTitle="blank").legendPositionBottom() +
    ggtitle("Five rows legend and below") + ggsize(700,400)

Out[9]:

In [10]:

// Set color to city MPG and shape to drive·train
val p1 = letsPlot(mpg) {x="displ"; y="hwy"} + geomPoint(size=5.0) {color="cty"; shape="drv"} + ggsize(700, 350)
p1

Out[10]:

In [11]:

// Change legend position and orientation
p1 + theme().legendPosition(1, 1).legendJustification(1, 1).legendDirectionHorizontal()

Out[11]:

In [12]:

// Adjust colorbar size
// Change order in drive-train legend
// Set labels to legends and axis
// Remove vertical axis line
p1 + theme().legendPosition(1, 1).legendJustification(1, 1).legendDirectionHorizontal().axisLineYBlank() +
    scaleColorContinuous(name="City MPG", 
                         low="dark_blue", high="light_blue", 
                         guide=guideColorbar(barHeight=10, barWidth=300)) +
    scaleShape(name="Drive-train", breaks=listOf('f', 'r', '4'), labels=listOf("front", "rear", "4X4")) +
    xlab("Engine displacement (L)") + ylab("Highway MPG")

Out[12]: