How to create a cmyk image in python

Is it possible to create a cmyk image using python like a rgb image? suppose we have an 2d array and each element is a 4-dim vector! How can we convert it to a cmyk image?

Have you looked at the Pillow library for images?

from PIL import Image
im = Image.fromarray(A, mode="CMYK")
im.save("your_file.jpeg")