如何在 Swift 中仅对数组的一部分求和？

Question

我目前正在尝试在 Swift 中构建我的第一个应用程序，并且正在寻找相对于当前月份的数组部分的总和。这是我的代码：

struct Hour {
    var date: String?
    var time: String?

    init(date: String?, time: String?) {
        self.date = date
        self.time = time
    }
}


let hoursData = [
    Hour(date: "Nov 29, 2015", time: "7"),
    Hour(date: "Dec 12, 2015", time: "7"),
    Hour(date: "Dec 14, 2015", time: "7"),
    Hour(date: "Dec 25, 2015", time: "7") ]

我想知道如何创建一个包含当月 time 数据总和的变量？或者任何一个月？你们能提供的任何帮助将不胜感激。

Answer 1

首先我会稍微重写一下你的结构，
这两个属性都可以是不可变的。

我使用 NSDate 代替字符串作为日期，并使用双精度表示持续时间

struct Hour {
    let date: NSDate
    let duration: Double

    init(date: NSDate, duration: Double) {
        self.date = date
        self.duration = duration
    }
}

我创造了像

这样的时间

let hoursData = [
    Hour(date: { let c = NSDateComponents(); c.day = 29; c.month = 11; c.year = 2015;  return NSCalendar.currentCalendar().dateFromComponents(c)!}(), duration: 7),
    Hour(date: { let c = NSDateComponents(); c.day = 12; c.month = 12; c.year = 2015;  return NSCalendar.currentCalendar().dateFromComponents(c)!}(), duration: 7),
    Hour(date: { let c = NSDateComponents(); c.day = 14; c.month = 12; c.year = 2015;  return NSCalendar.currentCalendar().dateFromComponents(c)!}(), duration: 7),
    Hour(date: { let c = NSDateComponents(); c.day = 15; c.month = 12; c.year = 2015;  return NSCalendar.currentCalendar().dateFromComponents(c)!}(), duration: 7)]

如果你想知道这个语法：我使用隐式未命名的闭包来创建 NSDate 参数

现在我过滤月份和年份，并使用reduce 方法来汇总过滤后的对象

let today = NSDate()
let totalDuration = hoursData.filter{
    let objectsComps = NSCalendar.currentCalendar().components([.Month, .Year], fromDate: $0.date)
    let todaysComps  = NSCalendar.currentCalendar().components([.Month, .Year], fromDate: today)
    return objectsComps.month == todaysComps.month && objectsComps.year == todaysComps.year
}.reduce(0) { (duration, hour) -> Double in
    duration + hour.duration
}

---

虽然这是一个很好的答案，但我想指出 Rob 的答案有一个小缺陷：Hour 结构隐藏了它对日期字符串需要具有某种格式的依赖性。这违反了Dependency Inversion Principle，SOLID Principles 中的 D。但这很容易解决。

创建Hour 实例时只需传入日期格式化程序。

struct Hour {
    let date: NSDate
    let duration: Double

    init(date: NSDate, duration: Double) {
        self.date = date
        self.duration = duration
    }

    init (dateFormatter:NSDateFormatter, dateString:String, duration:Double) {
        self.init(date: dateFormatter.dateFromString(dateString)!, duration:duration)
    }
}

let dateFormatter = NSDateFormatter()
dateFormatter.dateFormat = "MMM d, y"

let hoursData = [
    Hour(dateFormatter: dateFormatter, dateString: "Nov 29, 2015", duration: 7),
    Hour(dateFormatter: dateFormatter, dateString: "Dec 12, 2015", duration: 7),
    Hour(dateFormatter: dateFormatter, dateString: "Dec 14, 2015", duration: 7),
    Hour(dateFormatter: dateFormatter, dateString: "Dec 25, 2015", duration: 7),
    Hour(dateFormatter: dateFormatter, dateString: "Dec 25, 2017", duration: 7)
]

现在使用Hour 的人可以根据需要定义格式，这可能对本地化应用有所帮助。

过滤和减少保持不变。

---

但现在我们有一个新问题：NSDateformatter's dateFromString() 可能返回 nil。目前我们使用! 强制解包，但这在生产应用程序中可能很糟糕。

我们应该允许适当的错误处理，允许方便的 init 抛出错误

enum HourError: ErrorType {
    case InvalidDate
}

struct Hour {

    let date: NSDate
    let duration: Double

    init(date: NSDate, duration: Double) {
        self.date = date
        self.duration = duration
    }

    init (dateFormatter:NSDateFormatter, dateString:String, duration:Double) throws {
        let date = dateFormatter.dateFromString(dateString)
        guard date != nil else { throw HourError.InvalidDate}
        self.init(date: date!, duration:duration)
    }
}

如果我们像这样使用它

do {
    let now = NSDate()
    let dateFormatter = NSDateFormatter()
    dateFormatter.dateFormat = "MMM d, y"

    let hoursData = [
        try Hour(dateFormatter: dateFormatter, dateString: "Nov 29, 2015", duration: 7),
        try Hour(dateFormatter: dateFormatter, dateString: "Dec 12, 2015", duration: 7),
        try Hour(dateFormatter: dateFormatter, dateString: "Dec 14, 2015", duration: 7),
        try Hour(dateFormatter: dateFormatter, dateString: "Dec 25, 2015", duration: 7),
/*⛈*/  try Hour(dateFormatter: dateFormatter, dateString: " 25, 2017", duration: 7)
    ]



    let totalDuration = hoursData.filter{
        let objectsComps = NSCalendar.currentCalendar().components([.Month, .Year], fromDate: $0.date)
        let todaysComps  = NSCalendar.currentCalendar().components([.Month, .Year], fromDate: now)
        return objectsComps.month == todaysComps.month && objectsComps.year == todaysComps.year
        }.reduce(0) {
            $0 + $1.duration
    }
    print(totalDuration)
} catch HourError.InvalidDate{
    print("one or more datestrings must be wrong")
}

错误将被捕获。

---

full code

Answer 2

我可能会这样做：

// get prefix for current month

let formatter = NSDateFormatter()
formatter.dateFormat = "MMM"
let monthString = formatter.stringFromDate(NSDate())

// now add up the time values for that month

let results = hoursData.filter { $0.date?.hasPrefix(monthString) ?? false }
    .reduce(0) { $0 + (Int($1.time ?? "0") ?? 0) }

或者，如果您也想添加年份支票：

let formatter = NSDateFormatter()
formatter.dateFormat = "MMM"
let monthString = formatter.stringFromDate(NSDate())

formatter.dateFormat = "yyyy"
let yearString = formatter.stringFromDate(NSDate())

let results = hoursData.filter { $0.date?.hasPrefix(monthString) ?? false && $0.date?.hasSuffix(yearString) ?? false }
    .reduce(0) { $0 + (Int($1.time ?? "0") ?? 0) }

print(results)

注意，在上述两种情况下，由于您使用的是可选项，因此我使用 ?? 来处理值为 nil 的情况（以及 time 中是否存在非数字字符串） )。

---

就个人而言，我建议Hour 使用NSDate 和Float 而不是String，除非你真的需要它们，否则不要使用选项，并使用let 而不是var：

struct Hour {
    let date: NSDate
    let time: Float

    init(dateString: String, time: Float)  {
        let formatter = NSDateFormatter()
        formatter.dateFormat = "MMM, d, y"
        self.date = formatter.dateFromString(dateString)!
        self.time = time
    }
}

那么代码就变成了：

let hoursData = [
    Hour(dateString: "Nov 29, 2015", time: 7),
    Hour(dateString: "Dec 12, 2015", time: 7),
    Hour(dateString: "Dec 14, 2015", time: 7),
    Hour(dateString: "Dec 25, 2015", time: 7),
    Hour(dateString: "Dec 25, 2017", time: 7)
]

let calendar = NSCalendar.currentCalendar()
let currentComponents = calendar.components([.Month, .Year], fromDate: NSDate())

let results = hoursData.filter {
    let components = calendar.components([.Month, .Year], fromDate: $0.date)
    return components.month == currentComponents.month && components.year == currentComponents.year
}.reduce(0) { return $0 + $1.time }

print(results)

还有更多可能的优化（例如，不重复实例化NSDateFormatter），但它说明了使用NSDate 对象可以让您使用日历计算，而不是寻找子字符串。

Answer 3

你也可以在功能上做到这一点

let sum = hoursData
    .filter { $0.date?.hasPrefix("Dec") ?? false } // Only get dates in the correct month
    .flatMap { $0.time }                           // Map to an array of non nil times
    .flatMap { Int($0) }                           // Convert strings to ints
    .reduce(0) { $0 + $1 }                         // Sum up the times

如果您存储非可选的Ints 而不是可选的Strings，这会简单得多。或者你可以使用NSDateComponents。