Swift中无需编写循环即可基于Set保留Dictionary指定键的可行方案

嘿，这个需求很常见！确实Swift标准库没有直接提供类似Java Map.keySet().retainAll() 的现成API，但我们可以用更简洁的函数式写法替代手动for循环，让代码更优雅。

下面分享几种实用的实现方式：

方式1：生成新的过滤字典（推荐，无副作用） #

如果不需要修改原字典，而是想要一个只保留指定键的新字典，直接用filter方法最简洁：

let goodSet: Set = ["good_key_1", "good_key_2", "good_key_3"]
let filteredDict = dictionary.filter { goodSet.contains($0.key) }

这个方法是纯函数式的，不会改变原字典，代码清晰易懂，适合大多数场景。

方式2：直接修改原字典（批量删除） #

如果必须修改原有的可变字典（用var声明的），可以先筛选出需要删除的键，再批量移除：

let goodSet: Set = ["good_key_1", "good_key_2", "good_key_3"]
let keysToRemove = dictionary.keys.filter { !goodSet.contains($0) }
keysToRemove.forEach { dictionary[$0] = nil }

相比手动遍历判断删除，这种写法把逻辑拆分成两步，可读性更强，也避免了遍历过程中修改字典可能带来的潜在问题（虽然Swift允许这么做，但这种写法更规范）。

方式3：高效初始化新字典（适合大数据量） #

如果字典规模较大，想要更高效的实现，可以结合lazy和字典初始化器，避免额外的数组内存开销：

let goodSet: Set = ["good_key_1", "good_key_2", "good_key_3"]
let retainedDict = Dictionary(uniqueKeysWithValues: dictionary.lazy.filter { goodSet.contains($0.key) })

lazy会让过滤操作延迟执行，只在初始化字典时按需处理键值对，性能更优。

这些方法都不需要手动编写for循环的判断逻辑，更符合Swift的编程风格，根据你的实际需求选择就行～

内容的提问来源于stack exchange，提问作者Cheok Yan Cheng