Swift 4:
ตั้งแต่ Swift 4 ฟังก์ชันนี้ได้รับการ การเพิ่มในห้องสมุดมาตรฐาน คุณสามารถใช้มันดังนี้:
Dictionary(grouping: statEvents, by: { $0.name })
[
"dinner": [
StatEvents(name: "dinner", date: "01-01-2015", hours: 1),
StatEvents(name: "dinner", date: "01-01-2015", hours: 1),
StatEvents(name: "dinner", date: "01-01-2015", hours: 1)
],
"lunch": [
StatEvents(name: "lunch", date: "01-01-2015", hours: 1),
StatEvents(name: "lunch", date: "01-01-2015", hours: 1)
]
Swift 3:
public extension Sequence {
func group<U: Hashable>(by key: (Iterator.Element) -> U) -> [U:[Iterator.Element]] {
var categories: [U: [Iterator.Element]] = [:]
for element in self {
let key = key(element)
if case nil = categories[key]?.append(element) {
categories[key] = [element]
}
}
return categories
}
}
น่าเสียดายที่append
ฟังก์ชั่นด้านบนคัดลอกอาร์เรย์ที่อยู่ข้างใต้แทนที่จะเปลี่ยนรูปแบบแทนซึ่งจะดีกว่า ซึ่งทำให้การชะลอตัวค่อนข้างมากค่อนข้างมาก คุณสามารถแก้ไขปัญหาได้โดยใช้ Wrapper ประเภทการอ้างอิง:
class Box<A> {
var value: A
init(_ val: A) {
self.value = val
}
}
public extension Sequence {
func group<U: Hashable>(by key: (Iterator.Element) -> U) -> [U:[Iterator.Element]] {
var categories: [U: Box<[Iterator.Element]>] = [:]
for element in self {
let key = key(element)
if case nil = categories[key]?.value.append(element) {
categories[key] = Box([element])
}
}
var result: [U: [Iterator.Element]] = Dictionary(minimumCapacity: categories.count)
for (key,val) in categories {
result[key] = val.value
}
return result
}
}
แม้ว่าคุณจะสำรวจพจนานุกรมสุดท้ายสองครั้ง แต่เวอร์ชันนี้ก็ยังเร็วกว่าต้นฉบับในกรณีส่วนใหญ่
Swift 2:
public extension SequenceType {
func categorise<U : Hashable>(@noescape keyFunc: Generator.Element -> U) -> [U:[Generator.Element]] {
var dict: [U:[Generator.Element]] = [:]
for el in self {
let key = keyFunc(el)
if case nil = dict[key]?.append(el) { dict[key] = [el] }
}
return dict
}
}
ในกรณีของคุณคุณสามารถส่งคืน "คีย์" ได้โดยkeyFunc
เป็นชื่อ:
currentStat.statEvents.categorise { $0.name }
[
dinner: [
StatEvents(name: "dinner", date: "01-01-2015", hours: 1),
StatEvents(name: "dinner", date: "01-01-2015", hours: 1),
StatEvents(name: "dinner", date: "01-01-2015", hours: 1)
], lunch: [
StatEvents(name: "lunch", date: "01-01-2015", hours: 1),
StatEvents(name: "lunch", date: "01-01-2015", hours: 1)
]
]
ดังนั้นคุณจะได้รับพจนานุกรมโดยที่ทุกคีย์เป็นชื่อและทุกค่าคืออาร์เรย์ของ StatEvents ที่มีชื่อนั้น
สวิฟต์ 1
func categorise<S : SequenceType, U : Hashable>(seq: S, @noescape keyFunc: S.Generator.Element -> U) -> [U:[S.Generator.Element]] {
var dict: [U:[S.Generator.Element]] = [:]
for el in seq {
let key = keyFunc(el)
dict[key] = (dict[key] ?? []) + [el]
}
return dict
}
categorise(currentStat.statEvents) { $0.name }
ซึ่งให้ผลลัพธ์:
extension StatEvents : Printable {
var description: String {
return "\(self.name): \(self.date)"
}
}
print(categorise(currentStat.statEvents) { $0.name })
[
dinner: [
dinner: 01-01-2015,
dinner: 01-01-2015,
dinner: 01-01-2015
], lunch: [
lunch: 01-01-2015,
lunch: 01-01-2015
]
]
(swiftstub อยู่ที่นี่ )
Dictionary
init(grouping:by:)
เริ่มต้นใหม่