如何统计Core Data中Transformable类型数组数组的元素出现次数？

嘿，刚好碰到过类似的Core Data场景，给你梳理下最高效的实现方式：

你的colorArray是二维数组结构，首先得把它"摊平"成一维数组，这样才能方便遍历每一个颜色元素，然后用Swift的高阶函数或者NSCountedSet来统计次数，这两种方式都是线性时间复杂度，效率拉满。

方式一：在Core Data实体子类里封装统计方法（推荐） #

假设你的Core Data实体叫ColorEntry，对应的子类里colorArray是[[String]]类型（如果是自定义颜色枚举，只要符合Hashable就行，Swift枚举默认支持）：

import CoreData

class ColorEntry: NSManagedObject {
    @NSManaged var colorArray: [[String]]?
    
    // 直接在实体类里加统计方法，调用起来特别方便
    func calculateColorOccurrences() -> [String: Int] {
        guard let nestedColors = colorArray else { return [:] }
        
        // 第一步：把二维数组扁平化
        let allColors = nestedColors.flatMap { $0 }
        
        // 第二步：用reduce快速统计次数
        return allColors.reduce(into: [:]) { counts, color in
            counts[color, default: 0] += 1
        }
    }
}

用的时候就这么简单：

// 先从Core Data拿到你的实体实例
let colorEntry: ColorEntry = // 你的获取逻辑
let colorCounts = colorEntry.calculateColorOccurrences()
// 结果就是 ["Red": 3, "Blue": 4, "Green": 4]

方式二：用NSCountedSet快速实现 #

如果你想要更简洁的代码，或者需要和Objective-C兼容，NSCountedSet是个省心的选择，它会自动帮你统计元素出现次数：

func countColors(from colorArray: [[String]]?) -> [String: Int] {
    guard let nestedColors = colorArray else { return [:] }
    let flattened = nestedColors.flatMap { $0 }
    let countedSet = NSCountedSet(array: flattened)
    
    // 转成字典格式返回
    var result: [String: Int] = [:]
    for color in countedSet {
        if let colorStr = color as? String {
            result[colorStr] = countedSet.count(for: color)
        }
    }
    return result
}

为什么这两种方式最优？ #

• 扁平化数组：flatMap是Swift处理二维数组转一维的高效方法，时间复杂度是O(n)（n是所有颜色元素的总数）。
• reduce统计：reduce(into:)构建字典的过程也是线性的，整个流程下来没有多余的循环，效率最高。
• NSCountedSet：底层基于哈希表实现，统计效率同样是O(n)，而且代码更简洁，适合快速开发。

小提示 #

• 如果你的颜色是自定义枚举，确保它实现了Hashable（Swift枚举默认就支持，只要关联类型也符合要求），不然没法当字典的key或者放进NSCountedSet里。
• 一定要处理colorArray为nil的情况，避免崩溃，返回空字典是合理的默认行为。

内容的提问来源于stack exchange，提问作者user3227546