如何使用matplotlib创建圆形2D图，其中函数取决于距图像中心的距离？

import numpy as np
import matplotlib.pyplot as plt

size = 100  # 图像大小
image = np.zeros((size, size))  # 创建一个全零的二维数组

center = size // 2  # 图像中心坐标
x, y = np.meshgrid(range(size), range(size))  # 创建坐标网格
distance = np.sqrt((x - center) ** 2 + (y - center) ** 2)  # 计算距离

radius = size // 2  # 圆的半径
image[distance <= radius] = 1  # 距离小于等于半径的像素点置为1

plt.imshow(image, cmap='gray')  # 使用灰度色彩映射
plt.axis('off')  # 关闭坐标轴
plt.show()  # 显示图像