如何从pyspark dataframe中提取图像到numpy

from pyspark.sql.functions import udf
from PIL import Image
import numpy as np

def image_to_array(image_data):
    image = Image.open(io.BytesIO(image_data))
    return np.array(image)

image_to_array_udf = udf(image_to_array, ArrayType(ByteType()))

df = df.withColumn('image_array', image_to_array_udf('image_data'))

from pyspark.sql.functions import udf
from PIL import Image
import numpy as np

def image_to_array(image_data):
    image = Image.open(io.BytesIO(image_data))
    return np.array(image)

image_to_array_udf = udf(image_to_array, ArrayType(ByteType()))

df = df.withColumn('image_array', image_to_array_udf('image_data'))