ios - 如何按星期几对数据进行分组和求和？

Question

我有两个数组，保存日期的 nsDates 和保存值 (Int) 的 values，我想根据星期几对值求和.

我找到了几个关于 SQL 的答案，但在 swift/objective-c 中没有找到。我在下面编写的代码可以解决问题。但它很慢而且冗长。根据星期几对值进行分组和求和的计算最有效的方法是什么？任何帮助将不胜感激!

我创建了一个函数来获取星期几(即星期日 = 1 等)

func getDayOfWeek(somedate:NSDate)->Int? {

    let calendar:NSCalendar = NSCalendar.currentCalendar()
    let dateComps:NSDateComponents = calendar.components(.CalendarUnitWeekday , fromDate: somedate)
    let dayOfWeek:Int = dateComps.weekday
    return dayOfWeek
}

我初始化数组以保存我将要传递的值。

var friday = [Int]()
var saturday = [Int]()
var sunday = [Int]()
var monday = [Int]()
var tuesday = [Int]()
var wednesday = [Int]()
var thursday = [Int]()

我将 nsDates 中的日期传递给函数 getDayOfWeek，以获取星期几，如果它是 1，那么我将值数组中的相应值附加到我在上面初始化的星期日数组中，如果 dayOfWeek == 2我将它附加到星期一数组等。

func appendValuesPerDayOfWeek(){
    for i in 1...nsDates.count{
      var dayOfWeek = getDayOfWeek(nsDates[i-1])
            if dayOfWeek == 1{
                sunday.append(values[i-1])}
            if dayOfWeek == 2{
                monday.append(values[i-1])}
            if dayOfWeek == 3{
                tuesday.append(values[i-1])}
            if dayOfWeek == 4{
                wednesday.append(values[i-1])}
            if dayOfWeek == 5{
                thursday.append(values[i-1])}
            if dayOfWeek == 6{
                friday.append(values[i-1])}
            if dayOfWeek == 7{
                saturday.append(values[i-1])}
        }
}

我执行函数

appendValuesPerDayOfWeek()

为了获得一周中每一天的值总和，我最后对每个数组的元素求和。

let sumMonday = monday.reduce(0,+)
let sumTuesday = tuesday.reduce(0,+)
let sumWednesday = wednesday.reduce(0,+)
let sumThursday = thursday.reduce(0,+)
let sumFriday = friday.reduce(0,+)
let sumSaturday = saturday.reduce(0,+)
let sumSunday = sunday.reduce(0,+)

Answer 1

与其在一周中的每一天都有单独的变量，不如使用将一周中的几天映射到值的字典会更好。

鉴于此更改，您可以编写一个函数 groupedReduce，它同时采用组合函数和分组函数，并返回一个字典，其中每个组都组合在一起:

func groupedReduce
  <S: SequenceType, K: Hashable, U>
  (source: S, initial: U, 
   combine: (U, S.Generator.Element) -> U, 
   groupBy: (S.Generator.Element) -> K) 
  -> [K:U] 
{

    var result: [K:U] = [:]

    for element in source {
        let key = groupBy(element)
        result[key] = combine(result[key] ?? initial, element)
    }

    return result
}

let a = [1,5,2,3,7]
let isEven = { $0 % 2 == 0 }
groupedReduce(a, 0, +, isEven)
// returns dictionary of two groupings, false (odd numbers)
// summing to 16 and true (even numbers) summing to 2

这应该很容易适应采用按星期几分组的函数，并执行您正在寻找的任何求和逻辑。

请注意，它们的键必须是 Hashable。您的 getDayOfWeek 函数返回一个 Int? 并且可选值不可哈希——但它看起来不需要返回一个可选值。给定日期永远不会是星期几，因此它应该只返回一个 Int，并且它们 是可哈希的。