如何在Swift中保存按钮图像状态？

var isButtonSelected = false

@IBAction func buttonTapped(_ sender: UIButton) {
    if isButtonSelected {
        sender.setImage(UIImage(named: "unselectedImage"), for: .normal)
        isButtonSelected = false
    } else {
        sender.setImage(UIImage(named: "selectedImage"), for: .normal)
        isButtonSelected = true
    }
}

// 保存按钮状态
UserDefaults.standard.set(isButtonSelected, forKey: "buttonSelected")

// 读取按钮状态
if let buttonSelected = UserDefaults.standard.value(forKey: "buttonSelected") as? Bool {
    isButtonSelected = buttonSelected
    if isButtonSelected {
        button.setImage(UIImage(named: "selectedImage"), for: .normal)
    } else {
        button.setImage(UIImage(named: "unselectedImage"), for: .normal)
    }
}