diff -r 9b5165b8a762 -r 0587ef444547 Attic/alcohol.scala --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/Attic/alcohol.scala Sat Mar 11 23:22:05 2023 +0000 @@ -0,0 +1,72 @@ +// Part 2 about Alcohol-Consumption Worldwide +//============================================ + +object CW6b { + +import io.Source +import scala.util._ + +def get_csv_page(url: String) : List[String] = + Source.fromURL(url)("ISO-8859-1").getLines.toList + +def get_csv_file(file: String) : List[String] = + Source.fromFile(file)("ISO-8859-1").getLines.toList + + +val url_alcohol = + "https://raw.githubusercontent.com/fivethirtyeight/data/master/alcohol-consumption/drinks.csv" + +val file_population = + "population.csv" + +get_csv_page(url_alcohol) +get_csv_file(file_population) + +get_csv_page(url_alcohol).size +get_csv_file(file_population).size + +val alcs = get_csv_page(url_alcohol) +val pops = get_csv_file(file_population) + + +def process_alcs(lines: List[String]) : List[(String, Double)] = + for (l <- lines) yield { + val entries = l.split(",").toList + (entries(0), entries(4).toDouble) + } + +def process_pops(lines: List[String]) : Map[String, Long] = + (for (l <- lines) yield { + val entries = l.split(",").toList + (entries(0), entries(1).toLong) + }).toMap + + +def sorted_country_consumption() : List[(String, Long)] = { + val alcs2 = process_alcs(alcs.drop(1)) + val pops2 = process_pops(pops.drop(1)) + val cons_list = + for ((cname, cons) <- alcs2; + if pops2.isDefinedAt(cname)) yield (cname, (cons * pops2(cname)).toLong) + cons_list.sortBy(_._2).reverse +} + +sorted_country_consumption()(9) +sorted_country_consumption().size + +def percentage(n: Int) : (Long, Long, Double) = { + val cons_list = sorted_country_consumption() + val sum_n = cons_list.take(n).map(_._2).sum + val sum_all = cons_list.map(_._2).sum + val perc = (sum_n.toDouble / sum_all.toDouble) * 100.0 + (sum_all, sum_n, perc) +} + + +percentage(10) +percentage(164) + +assert(percentage(164) == (28771558364L, 28771558364L, 100.0)) + +} +