集合视图中的自定义布局

Question

uicollectionview Pinterest布局中可能存在多个部分。 Pinterest Layout类中有多个部分正在工作。我已经尝试过使用Pinterest布局类。请帮忙。

func numberOfSections(in collectionView: UICollectionView) -> Int {
        return 2
    }

    func collectionView(_ collectionView: UICollectionView, numberOfItemsInSection section: Int) -> Int {
        if section == 0 {
            return 4
        }else {
            return 20
        }
    }