Wolfram Language Paclet Repository
Community-contributed installable additions to the Wolfram Language
Mosaic plots for numerical variables via categorical mapping
In this notebook we show how to transform numerical variables into categorical in order to get more informative mosaic plots.
Load the paclet
In[52]:=
Needs["AntonAntonov`MosaicPlot`"]
Here we get the Titanic dataset:
In[53]:=
dsTitanic=ResourceFunction["ExampleDataset"][{"MachineLearning","Titanic"}]
Out[53]=
| |||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
Here is a mosaic plot of passenger class vs. passenger age:
In[54]:=
 | MosaicPlot |
Out[54]=
The plot is not very informative
Here we make a mapping function of the numerical variable "passenger age" into "categorical age" integers (factors):
In[55]:=
factor=Piecewise[{{1,-Infinity<##1≤5},{2,5<##1≤14},{3,14<##1≤21},{4,21<##1≤28},{5,28<##1≤35},{6,35<##1≤50},{7,50<##1≤Infinity}},0]&;
Here we make a list of rules that map the categorical integers (factors) into (informative) labels:
In[56]:=
aFactorLabel={1"1(under 6)",2"2(6...14)",3"3(15...21)",4"4(22...]28)",5"5(29...35)",6"6(36...50)",7"7(50+)",0"0(missing)"};
Here we add the column "AgeFactor" that has the mapped values of "passenger age":
In[57]:=
dsTitanic2=dsTitanic[All,#,"AgeFactor"If[Head[#["passenger age"]]===Missing,0,factor[#["passenger age"]]]/.aFactorLabel&];
Here is the mosaic plot of passenger class vs. categorical passenger age:
In[59]:=
| MosaicPlot |
Out[59]=
The second mosaic plot is more informative.
""