swift 如何把 颜色直接换换成图片

func creatImageWithColor(color:UIColor)->UIImage{
        let rect = CGRectMake(0.0,0.0,1.0,1.0)
        UIGraphicsBeginImageContext(rect.size)
        let context = UIGraphicsGetCurrentContext()
        CGContextSetFillColorWithColor(context,color.CGColor)
        CGContextFillRect(context,rect)
        let image = UIGraphicsGetImageFromCurrentImageContext()
        UIGraphicsEndImageContext()
        return image
    }