按SWIFT中每个ID的最大值进行筛选排序

发布于12月12日

我有一个对象数组，其 struct 如下:

{
  id: String,
  userId: String,
  distance: Double,
  time: Double
}

在数组中，相同的userId在相同的距离内可以有多个条目.例如，数组可以是:

[
  { id: "1", userId: "1", distance: 100, time: 18.7 },
  { id: "2", userId: "1", distance: 100, time: 18.5 }, 
  { id: "3", userId: "2", distance: 100, time: 18.2 },
  { id: "4", userId: "2", distance: 200, time: 41.0 },
]

我想对数组进行筛选和排序，以便该数组只包含每个用户特定距离的最快时间.因此，如果我想使用上面的数组来查找100次，它将简单地返回:

[
  { id: "3", userId: "2", distance: 100, time: 18.2 },
  { id: "2", userId: "1", distance: 100, time: 18.5 },
]

我知道我可以用.filter来把它减少到.filter次，用.sort来安排时间，但我想不出怎么才能限制到每userId条只有1条.

推荐答案

您可以使用我添加到swift-algorithms(Apple托管的第一方包，就像Swift标准库的附件)的新grouped(by:)函数来实现这一点.

我们可以根据它们的距离和跑它们的用户来分组所有的比赛结果.在每组结果中，我们将 Select 一个最快的时间(该用户在该距离内).我把这个叫做userPersonalRecords.

从那里，我们可以根据距离进行筛选，以获得每个人最好的100米冲刺.

如果我们经常这样做，可能值得按距离对比赛结果进行分组，并保留结果array.或者，如果您有lots个结果，您可以将所有这些结果转储到一个小的SQLite DB中，并根据需要对查询执行相同的简单分组/排序.

struct RaceResult {
    let id: String
    let userId: String
    let distance: Double
    let time: Double
}

let raceResults = [
    RaceResult(id: "1", userId: "1", distance: 100, time: 18.7),
    RaceResult(id: "2", userId: "1", distance: 100, time: 18.5),
    RaceResult(id: "3", userId: "2", distance: 100, time: 18.2),
    RaceResult(id: "4", userId: "2", distance: 200, time: 41.0),
]

// This simple little struct is just a workaround for tuples not being Hashable
struct UserDistance: Hashable {
    let userId: String
    let distance: Double
}

let userPersonalRecords = raceResults
    .grouped {
        UserDistance(userId: $0.userId, distance: $0.distance)
    }
    .mapValues { raceResults in raceResults.max(by: { $0.time < $1.time })! }
    .values

let fastest100MTimes = userPersonalRecords.filter { $0.distance == 100 }

for raceResult in fastest100MTimes {
    print(raceResult)
}