Сложные многомерные список операций в Scala

Question

Учитывая список таких как следующее:

val dane = List(
    ("2011-01-04", -137.76), 
    ("2011-01-04", 2376.45), 
    ("2011-01-04", -1.70), 
    ("2011-01-04", -1.70), 
    ("2011-01-04", -1.00), 
    // ... skip a few ... 
    ("2011-12-22", -178.02), 
    ("2011-12-29", 1800.82), 
    ("2011-12-23", -83.97), 
    ("2011-12-24", -200.00), 
    ("2011-12-24", -30.55), 
    ("2011-12-30", 728.00) 
) 

Я хотел бы суммировать значения (то есть второй элемент внутренних списков) определенного месяца (например, январь, или 01), используя следующие операции в заданном порядке:

1. groupBy
2. slice
3. collect
4. sum

Answer 1

Я чувствую, наоборот, так вот ответ, который не использует ни один из установленных способов: groupBy, slice, collect или sum

Избежание collect была самая трудная часть, condOpt/flatten просто так уродливее ...

val YMD = """(\d\d\d\d)-(\d\d)-(\d\d)""".r 

import PartialFunction._ 

(dane map { 
    condOpt(_:(String,Double)){ case (YMD(_,"01",_), v) => v } 
}).flatten reduceLeft {_+_} 

Answer 2

Перерыв в проблеме на более мелкие шаги. Начните с попытки разбить список на один список на каждый месяц. Вы можете использовать для этого groupBy. Ваша первая проблема, вероятно, будет заключаться в том, как разбирать строку даты. Общее решение - использовать пользовательский класс даты и регулярное выражение; однако в этом контексте может быть целесообразным упрощенное решение с использованием индексированной подстроки (или slice).

Общий совет - загрузить данные в Scala REPL и поиграть с ним. Удачи.

Answer 3

Теперь, когда Кевин начал тенденцию противоположных ответов, вот тот, который вы никогда не должны использовать, но черт возьми, это работает! (И избегает каждый запрошенный метод, и будет работать на любой месяц, если вы измените строку, но это требует, что список будет отсортирован по дате.)

dane.scanLeft(("2011-01",0.0))((l,r) => 
    (l._1, 
    if ((l._1 zip r._1).forall(x => x._1==x._2)) l._2+r._2 else 0.0 
) 
).dropWhile(_._2==0).takeWhile(_._2 != 0.0).reverse.head._2 

Answer 4

import scala.collection.mutable.HashMap 
val totals = new HashMap[Int, Double] 
for (e <- dane) { 
    val (date, value) = e 
    val month = date.drop(5).take(2).toInt 
    totals(month) = totals.getOrElse(month,0.0) + value 
} 

Другая реализация не используя ни один из предложенных функций , а также изменчивые коллекции и некоторые ублюдочные сочетания процедурного и функционального стиля, избегая некоторых полезных функций :)

totals заканчивается как карта с номера месяца до общей суммы.

Answer 5

Я отказываюсь запутывать sum.

import org.joda.time.DateMidnight 
for (month <- 1 to 12) yield { 
    dane map { case (d,v) => new DateMidnight(d).getMonthOfYear -> v } 
    filter { case (m, v) => m == month } 
    map (_._2) 
    sum 
} 

Answer 6

(for((YearMonthDay(_, 1, _), value)<-dane) yield value).sum 

object YearMonthDay{ 
    def unapply(dateString:String):Option((Int, Int, Int)) ={ 
     //yes, there should really be some error checking in this extractor 
     //to return None for a bad date string 
     val components = dateString.split("-") 
     Some((components(0).toInt, components(1).toInt, components(2).toInt)) 
    } 

} 

Answer 7

Так вот идея:

• groupBy, потому что вы должны группировать данные от каждого месяца вместе
• slice, потому что вы должны видеть, что месяц из дата
• collect, поскольку необходимо filter по месяцам и map для оценки
• sum, мммм ... Я не уверен, где этот человек входит.Есть идеи?

Answer 8

dane.groupBy (_._1.matches (".*-01-.*")).slice (0, 1).map (x => x._2).flatten .map (y => y._2).sum 

Я действительно должен смотреть вверх «собирать», который так или иначе должен заменить мою карту/выравниваться/карту.

Мой результат: Double = 2234.29