Swift - pickerView组件的分隔符

class ViewController: UIViewController, UIPickerViewDelegate, UIPickerViewDataSource {

    let pickerView = UIPickerView()
    let options = ["Option 1", "Option 2", "Option 3", "Option 4", "Option 5"]
    let separatorIndexes = [2, 4] // 分隔符所在的行索引

    override func viewDidLoad() {
        super.viewDidLoad()

        pickerView.delegate = self
        pickerView.dataSource = self

        view.addSubview(pickerView)
    }

    // UIPickerViewDataSource协议方法
    func numberOfComponents(in pickerView: UIPickerView) -> Int {
        return 1 // 单列
    }

    func pickerView(_ pickerView: UIPickerView, numberOfRowsInComponent component: Int) -> Int {
        return options.count
    }

    // UIPickerViewDelegate协议方法
    func pickerView(_ pickerView: UIPickerView, viewForRow row: Int, forComponent component: Int, reusing view: UIView?) -> UIView {
        var customView: UIView

        if separatorIndexes.contains(row) {
            // 创建分隔符视图
            customView = UIView(frame: CGRect(x: 0, y: 0, width: pickerView.bounds.width, height: 1))
            customView.backgroundColor = UIColor.lightGray
        } else {
            // 创建其他选项视图
            let label = UILabel()
            label.text = options[row]
            label.textAlignment = .center
            customView = label
        }

        return customView
    }
}