如何在Python (Jupyter)中创建一个函数来计算点扩散函数(图像处理？)

import numpy as np
import matplotlib.pyplot as plt

def point_spread_function(image_size, point_location, point_intensity, sigma):
    x = np.arange(0, image_size[0], 1)
    y = np.arange(0, image_size[1], 1)
    xx, yy = np.meshgrid(x, y)
    psf = point_intensity * np.exp(-((xx - point_location[0])**2 + (yy - point_location[1])**2) / (2 * sigma**2))
    return psf

image_size = (512, 512)
point_location = (256, 256)
point_intensity = 1.0
sigma = 10.0

psf = point_spread_function(image_size, point_location, point_intensity, sigma)

plt.imshow(psf, cmap='gray')
plt.colorbar()
plt.title('Point Spread Function')
plt.show()