Stem-and-leaf plot
You are encouraged to solve this task according to the task description, using any language you may know.
Create a well-formatted stem-and-leaf plot from the following data set, where the leaves are the last digits:
110 436 124 109 440 330 53 352 315 452 54 49 334 102 432 123 442 125 97 104 11 446 123 360 324 427 451 329 139 42 324 320 450 100 87 414 305 21 375 324 360 123 33 378 37 66 41 321 68 356 407 448 5 128 81 361 419 134 147 146
The primary intent of this task is the presentation of information. It is acceptable to hardcode the data set or characteristics of it (such as what the stems are) in the example, insofar as it is impractical to make the example generic to any data set. For example, in a computation-less language like HTML the data set may be entirely prearranged within the example; the interesting characteristics are how the proper visual formatting is arranged.
If possible, the output should not be a bitmap image. Monospaced plain text is acceptable, but do better if you can. It may be a window, i.e. not a file.
Note: If you wish to try multiple data sets, you might try this generator.
Haskell
<lang haskell>import Data.List import Control.Arrow import Control.Monad
nls :: [Int] nls = [110, 436, 124, 109, 440, 330, 53, 352, 315, 452,
54, 49, 334, 102, 432, 123, 442, 125, 97, 104, 11, 446, 123, 360, 324, 427, 451, 329, 139, 42, 324, 320, 450, 100, 87, 414, 305, 21, 375, 324, 360, 123, 33, 378, 37, 66, 41, 321, 68, 356, 407, 448, 5, 128, 81, 361, 419, 134, 147, 146]
groupWith f = takeWhile(not.null). unfoldr(Just. (partition =<< (. f). (==). f. head)) justifyR = foldl ((. return) . (++) . tail) . flip replicate ' '
task ds = mapM_ (putStrLn. showStemLeaves justifyR fb. (head *** sort.concat). unzip)
$ groupWith fst $ stems ++ map (second return) stemLeaf where stemLeaf = map (`quotRem` 10) ds
stems = map (flip(,)[]) $ uncurry enumFromTo $ minimum &&& maximum $ fst $ unzip stemLeaf showStemLeaves f w (a,b) = f w (show a) ++ " |" ++ concatMap (f w. show) b fb = length $ show $ maximum $ map abs ds</lang> Output:
*Main> task nls 0 | 5 1 | 1 2 | 1 3 | 3 7 4 | 1 2 9 5 | 3 4 6 | 6 8 7 | 8 | 1 7 .... 37 | 5 8 38 | 39 | 40 | 7 41 | 4 9 42 | 7 43 | 2 6 44 | 0 2 6 8 45 | 0 1 2
J
Solution: (Tacit) <lang j>stemleaf =: 10&(<.@:%~ (~.@[ ; <@(</.)) |)@/:~ expandStems =: <./ ([ + i.@>:@-~) >./ expandLeaves=: (expandStems e. ])@[ #inv ]
showStemLeaf=: (":@,.@expandStems@[ ; ":&>@expandLeaves)&>/@stemleaf</lang>
Solution: (Explicit) <lang j>stemleafX=: monad define
dat=. /:~ y NB. sort data leaves=. 10 | dat stems=. dat <.@:% 10 leaves=. stems </. leaves NB. group leaves by stem (~.stems);<leaves
)
showStemLeafX=: monad define
'stems leaves'=. stemleafX y xstems=. (<./ ([ + i.@>:@-~ ) >./) stems NB. stems including those without leaves xleaves=. (xstems e. stems) #inv leaves NB. expand leaves to xstems (": ,.xstems) ; ":&> xleaves
)</lang>
Example: <lang j> nls =: ; <@(_&".);._2 noun define 110 436 124 109 440 330 53 352 315 452 54 49 334 102 432 123 442 125 97 104 11 446 123 360 324 427 451 329 139 42 324 320 450 100 87 414 305 21 375 324 360 123 33 378 37 66 41 321 68 356 407 448 5 128 81 361 419 134 147 146 )
showStemLeaf nls NB. display has been abbreviated
┌──┬───────────┐ │ 0│5 │ │ 1│1 │ │ 2│1 │ │ 3│3 7 │ │ 4│1 2 9 │ ... │42│7 │ │43│2 6 │ │44│0 2 6 8 │ │45│0 1 2 │ └──┴───────────┘</lang>
Perl generating LaTeX
<lang perl>#!/usr/bin/perl -w
my @data = sort {$a <=> $b} qw(110 436 124 109 440 330 53 352 315 452 54 49 334 102 432 123 442 125 97 104 11 446 123 360 324 427 451 329 139 42 324 320 450 100 87 414 305 21 375 324 360 123 33 378 37 66 41 321 68 356 407 448 5 128 81 361 419 134 147 146);
- FIXME: This should count the maximum number of leaves in any one stem;
- instead it takes the total number of data items, which is usually
- a massive overestimate.
my $columns = @data;
print <<"EOT"; \\documentclass{report} \\usepackage{fullpage} \\begin{document}
\\begin{tabular}{ r | *{$columns}{c} }
EOT
my $laststem = undef;
for my $value (@data) {
my $stem = int($value / 10); my $leaf = $value % 10; while (not defined $laststem or $stem > $laststem) { if (not defined $laststem) { $laststem = $stem - 1; } else { print " \\\\\n"; } $laststem++; print " $laststem"; } printf " & %3d", $leaf;
}
print <<'EOT';
\end{tabular}
\end{document} EOT</lang>
LaTeX output of the Perl program:
<lang latex>\documentclass{report} \usepackage{fullpage} \begin{document}
\begin{tabular}{ r | *{60}{c} } 0 & 5 \\ 1 & 1 \\ 2 & 1 \\ 3 & 3 & 7 \\ ... 44 & 0 & 2 & 6 & 8 \\ 45 & 0 & 1 & 2 \end{tabular}
\end{document}</lang>
The parameter to the tabular
environment defines the columns of the table. “r” and “c” are right- and center-aligned columns, “|” is a vertical rule, and “*{count}{cols}”
repeats a column definition count times.
Python
Adjusting Stem.leafdigits
allows you to modify how many digits of a value are used in the leaf, with the stem intervals adjusted accordingly.
<lang python>
from collections import namedtuple
from pprint import pprint as pp
from math import floor
Stem = namedtuple('Stem', 'data, leafdigits')
data0 = Stem((110, 436, 124, 109, 440, 330, 53, 352, 315, 452,
54, 49, 334, 102, 432, 123, 442, 125, 97, 104, 11, 446, 123, 360, 324, 427, 451, 329, 139, 42, 324, 320, 450, 100, 87, 414, 305, 21, 375, 324, 360, 123, 33, 378, 37, 66, 41, 321, 68, 356, 407, 448, 5, 128, 81, 361, 419, 134, 147, 146), 2.0)
def stemplot(stem):
d = [] interval = int(10**int(stem.leafdigits)) for data in sorted(stem.data): data = int(floor(data)) stm, lf = divmod(data,interval) d.append( (int(stm), int(lf)) ) stems, leafs = list(zip(*d)) stemwidth = max(len(str(x)) for x in stems) leafwidth = max(len(str(x)) for x in leafs) laststem, out = min(stems) - 1, [] for s,l in d: while laststem < s: laststem += 1 out.append('\n%*i |' % ( stemwidth, laststem)) out.append(' %0*i' % (leafwidth, l)) out.append('\n\nKey:\n Stem multiplier: %i\n X | Y => %i*X+Y\n' % (interval, interval)) return .join(out)
if __name__ == '__main__':
print( stemplot(data0) ) print( stemplot(Stem(data0.data, 1.0)) )</lang>
Sample Output
>>> 0 | 05 11 21 33 37 41 42 49 53 54 66 68 81 87 97 1 | 00 02 04 09 10 23 23 23 24 25 28 34 39 46 47 2 | 3 | 05 15 20 21 24 24 24 29 30 34 52 56 60 60 61 75 78 4 | 07 14 19 27 32 36 40 42 46 48 50 51 52 Key: Stem multiplier: 100 X | Y => 100*X+Y 0 | 5 1 | 1 2 | 1 3 | 3 7 4 | 1 2 9 5 | 3 4 6 | 6 8 7 | 8 | 1 7 9 | 7 10 | 0 2 4 9 11 | 0 12 | 3 3 3 4 5 8 13 | 4 9 14 | 6 7 15 | 16 | 17 | 18 | 19 | 20 | 21 | 22 | 23 | 24 | 25 | 26 | 27 | 28 | 29 | 30 | 5 31 | 5 32 | 0 1 4 4 4 9 33 | 0 4 34 | 35 | 2 6 36 | 0 0 1 37 | 5 8 38 | 39 | 40 | 7 41 | 4 9 42 | 7 43 | 2 6 44 | 0 2 6 8 45 | 0 1 2 Key: Stem multiplier: 10 X | Y => 10*X+Y >>>
Ruby
I added a few negative values to the data to demonstrate the "negative zero" stem. <lang ruby>class StemLeafPlot
def initialize(data, options = {}) opts = {:leaf_digits => 1}.merge(options) @leaf_digits = opts[:leaf_digits] @multiplier = 10 ** @leaf_digits @plot = generate_structure(data) end
private
def generate_structure(data) plot = Hash.new {|h,k| h[k] = []} data.sort.each do |value| stem, leaf = parse(value) plot[stem] << leaf end plot end
def parse(value) stem, leaf = value.abs.divmod(@multiplier) [Stem.get(stem, value < 0 ? :- : :+), leaf.round] end
public
def print stem_width = Math.log10(@plot.keys.max_by {|s| s.value}.value).ceil + 1 Stem.get_range(@plot.keys).each do |stem| leaves = @plot[stem].inject("") {|str,leaf| str << "%*d " % [@leaf_digits, leaf]} puts "%*s | %s" % [stem_width, stem, leaves] end
puts "key: 5|4=#{5 * @multiplier + 4}" puts "leaf unit: 1" puts "stem unit: #@multiplier" end
end
class Stem
@@cache = {}
def self.get(value, sign) if @@cachevalue, sign.nil? @@cachevalue, sign = self.new(value, sign) end @@cachevalue, sign end
def initialize(value, sign) @value = value @sign = sign end attr_accessor :value, :sign def negative? @sign == :- end
def <=>(other) if self.negative? if other.negative? other.value <=> self.value else -1 end else if other.negative? 1 else self.value <=> other.value end end end
def to_s "%s%d" % [(self.negative? ? '-' : ' '), @value] end def self.get_range(array_of_stems) min, max = array_of_stems.minmax if min.negative? if max.negative? min.value.downto(max.value).collect {|n| get(n, :-)} else min.value.downto(0).collect {|n| get(n, :-)} + 0.upto(max.value).collect {|n| get(n, :+)} end else min.value.upto(max.value).collect {|n| get(n, :+)} end end
end
data = DATA.read.split.map {|s| Float(s)} StemLeafPlot.new(data, :leaf_digits => 2).print puts StemLeafPlot.new(data).print
__END__
0 -3 -45 -167 110 436 124 109 440 330 53 352 315 452 54 49 334 102 432 123 442 125 97 104 11 446 123 360 324 427 451 329 139 42 324 320 450 100 87 414 305 21 375 324 360 123 33 378 37 66 41 321 68 356 407 448 5 128 81 361 419 134 147 146</lang>
outputs
-1 | 67 -0 | 45 3 0 | 0 5 11 21 33 37 41 42 49 53 54 66 68 81 87 97 1 | 0 2 4 9 10 23 23 23 24 25 28 34 39 46 47 2 | 3 | 5 15 20 21 24 24 24 29 30 34 52 56 60 60 61 75 78 4 | 7 14 19 27 32 36 40 42 46 48 50 51 52 key: 5|4=504 leaf unit: 1 stem unit: 100 -16 | 7 -15 | -14 | -13 | -12 | -11 | -10 | -9 | -8 | -7 | -6 | -5 | -4 | 5 -3 | -2 | -1 | -0 | 3 0 | 0 5 1 | 1 2 | 1 3 | 3 7 4 | 1 2 9 5 | 3 4 6 | 6 8 7 | 8 | 1 7 9 | 7 10 | 0 2 4 9 11 | 0 12 | 3 3 3 4 5 8 13 | 4 9 14 | 6 7 15 | 16 | 17 | 18 | 19 | 20 | 21 | 22 | 23 | 24 | 25 | 26 | 27 | 28 | 29 | 30 | 5 31 | 5 32 | 0 1 4 4 4 9 33 | 0 4 34 | 35 | 2 6 36 | 0 0 1 37 | 5 8 38 | 39 | 40 | 7 41 | 4 9 42 | 7 43 | 2 6 44 | 0 2 6 8 45 | 0 1 2 key: 5|4=54 leaf unit: 1 stem unit: 10
Tcl
Note that this algorithm collects the data in a hash table first before sorting the stems and then finally sorting the leaves by stem, rather than sorting the data first.
<lang tcl>package require Tcl 8.5
- How to process a single value, adding it to the table mapping stems to
- leaves.
proc addSLValue {arrayName value} {
upvar 1 $arrayName ary # Extract the sign and clean up the value set s [expr {$value < 0 ? "-" : " "}] set value [expr {round(abs($value))}] # Split the value into stem and leaf set leaf [expr {$value % 10}] set stem [expr {$value / 10}] # Store lappend ary($s$stem) $leaf
}
- How to do the actual output of the stem-and-leaf table, given that we have
- already done the splitting into stems and leaves.
proc printSLTable {arrayName} {
upvar 1 $arrayName ary # Sort the stems by number set names [lsort -integer [array names ary]] # Work out how much width the stems take so everything lines up set len [expr {
max([string length [lindex $names 0]], [string length [lindex $names end]])
}] # Print out the table, sorting the leaves as we go foreach n $names {
puts [format " %*s | %s" $len $n [lsort -integer $ary($n)]]
}
}
- Assemble the parts into a full stem-and-leaf table printer.
proc printStemLeaf dataList {
array set tbl {} foreach value $dataList {
addSLValue tbl $value
} printSLTable tbl
}
- Demo code
set data {
110 436 124 109 440 330 53 352 315 452 54 49 334 102 432 123 442 125 97 104 11 446 123 360 324 427 451 329 139 42 324 320 450 100 87 414 305 21 375 324 360 123 33 378 37 66 41 321 68 356 407 448 5 128 81 361 419 134 147 146
} printStemLeaf $data</lang> Output:
0 | 5 1 | 1 2 | 1 3 | 3 7 4 | 1 2 9 5 | 3 4 6 | 6 8 8 | 1 7 9 | 7 10 | 0 2 4 9 11 | 0 12 | 3 3 3 4 5 8 13 | 4 9 14 | 6 7 30 | 5 31 | 5 32 | 0 1 4 4 4 9 33 | 0 4 35 | 2 6 36 | 0 0 1 37 | 5 8 40 | 7 41 | 4 9 42 | 7 43 | 2 6 44 | 0 2 6 8 45 | 0 1 2