UICollectionView字典数据源下numberOfItemsInSection设置问题

这个问题我之前做项目时也踩过坑——字典的键是字符串，可UICollectionView的section参数是Int类型，直接对应不上确实头疼。给你分享两个实用的解决思路，看哪个更适配你的场景：

方法一：用有序数组存储字典键（最通用的动态场景） #

这是我最常用的方案，核心是把字典的字符串键提取成一个有序数组，作为section索引和字典键之间的映射表：

1. 先定义一个专门存section对应键的数组：

var sectionKeys: [String] = []

2. 在初始化数据源或者更新字典的时候，把字典的键转成数组存进去（如果需要固定顺序，记得加sorted()排序，比如按字母顺序或者业务逻辑排序）：

// 假设myDict是你的数据源字典
sectionKeys = Array(myDict.keys).sorted() // 排序可选，根据你的需求调整

3. 接下来实现UICollectionView的数据源方法就顺理成章了：

// 返回section数量
func numberOfSections(in collectionView: UICollectionView) -> Int {
    return sectionKeys.count
}

// 返回对应section的item数量
func collectionView(_ collectionView: UICollectionView, numberOfItemsInSection section: Int) -> Int {
    // 用section索引取对应的键，再从字典里拿数组的长度
    let currentKey = sectionKeys[section]
    // 加空值判断，避免字典里没有对应键时崩溃，返回0即可
    return myDict[currentKey]?.count ?? 0
}

注意：每次更新myDict后，一定要同步更新sectionKeys，然后调用collectionView.reloadData()，不然会出现索引不匹配的崩溃哦！

方法二：用枚举映射section与字符串键（适合固定section场景） #

如果你的字典键是固定不变的（比如只有几个固定分类），可以用枚举来做类型安全的映射：

1. 先定义一个对应section的枚举，每个case对应你的字典键：

enum CollectionSection: Int {
    case hotItems = 0
    case newItems = 1
    case recommendedItems = 2
    
    // 给每个枚举case绑定对应的字典键
    var dictKey: String {
        switch self {
        case .hotItems: return "hot"
        case .newItems: return "new"
        case .recommendedItems: return "recommend"
        }
    }
}

2. 然后实现数据源方法：

func numberOfSections(in collectionView: UICollectionView) -> Int {
    // 枚举case的数量就是section数
    return CollectionSection.allCases.count
}

func collectionView(_ collectionView: UICollectionView, numberOfItemsInSection section: Int) -> Int {
    // 把Int类型的section转成枚举，再取对应的字典键
    guard let currentSection = CollectionSection(rawValue: section) else {
        return 0
    }
    return myDict[currentSection.dictKey]?.count ?? 0
}

这个方案的好处是类型更安全，不用担心键名拼写错误，但缺点是没法动态适配字典键的增减，适合固定section的场景。

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