根据元素的出现获取数组的子集

Question

我有一个类似的数组：

var arr = [4,1,5,5,3]

我想根据其中元素的出现从数组中获取子集。

例如：

Elements with frequency 1 is {4,1,3}
Elements with frequency 2 is {5,5}

我关注了这个StackOverflow question，但无法弄清楚如何做上述事情。

有什么办法可以做到吗？

Answer 1

您可以使用NSCountedSet 来获取arr 中所有元素的计数，然后您可以构建Dictionary，其中键是元素的出现次数，值是数组具有关键出现次数的元素。通过迭代 Set(arr) 而不是简单地 arr 来构建字典，您可以确保重复元素只添加一次到字典中（例如，对于您的原始示例，5 不会添加两次，因为频率为 2)。

对于打印，您只需要遍历 Dictionary 的键并打印键及其对应的值。我只是对键进行排序，以使打印按出现次数的升序进行。

let arr = [4,1,5,5,3,2,3,6,2,7,8,2,7,2,8,8,8,7]
let counts = NSCountedSet(array: arr)
var countDict = [Int:[Int]]()
for element in Set(arr) {
    countDict[counts.count(for: element), default: []].append(element)
}
countDict

for freq in countDict.keys.sorted() {
    print("Elements with frequency \(freq) are {\(countDict[freq]!)}")
}

输出：

Elements with frequency 1 are {[4, 6, 1]}
Elements with frequency 2 are {[5, 3]}
Elements with frequency 3 are {[7]}
Elements with frequency 4 are {[2, 8]}

Swift 3 版本：

let arr = [4,1,5,5,3,2,3,6,2,7,8,2,7,2,8,8,8,7]
let counts = NSCountedSet(array: arr)
var countDict = [Int:[Int]]()
for element in Set(arr) {
    if countDict[counts.count(for: element)] != nil {
        countDict[counts.count(for: element)]!.append(element)  
    } else {
        countDict[counts.count(for: element)] = [element]
    }
}

for freq in countDict.keys.sorted() {
    print("Elements with frequency \(freq) are {\(countDict[freq]!)}")
}

Answer 2

您只需要获取元素的出现并过滤仅出现一次或多次的元素，如answer所示：

extension Array where Element: Hashable {
    // Swift 4 or later
    var occurrences: [Element: Int] {
        return reduce(into: [:]) { $0[$1, default: 0] += 1 }
    }
    // // for Swift 3 or earlier
    // var occurrences: [Element: Int] {
    //     var result: [Element: Int] = [:]
    //     forEach{ result[$0] = (result[$0] ?? 0) + 1}
    //     return result
    // }
    func frequencies(where isIncluded: (Int) -> Bool) -> Array {
        return filter{ isIncluded(occurrences[$0] ?? 0) }
    }
}

---

游乐场测试：

let arr = [5, 4, 1, 5, 5, 3, 5, 3]

let frequency1 = arr.frequencies {$0 == 1}       // [4, 1]
let frequency2 = arr.frequencies {$0 == 2}       // [3, 3]
let frequency3orMore = arr.frequencies {$0 >= 3} // [5, 5, 5, 5]

Answer 3

就是这样：

func getSubset(of array: [Int], withFrequency frequency: Int) -> [Int]
{
    var counts: [Int: Int] = [:]

    for item in array
    {
        counts[item] = (counts[item] ?? 0) + 1
    }

    let filtered  = counts.filter{ $0.value == frequency}

    return Array(filtered.keys)
}

这是纯 Swift（不使用旧的 Next Step 类），并且正在使用您提供的 SO 链接中的想法。

counts 字典包含数组中每个 int 值（键）的频率（值）：[int-value : frequency]。