Преобразовать объект numpy.ndarray в объект matplotlib.Axes

Question

У меня есть массив numpy (current_ax) формы (n, n), который состоит из matplotlib.AxesSubplot в каждом индексе.Я хочу преобразовать current_ax в объект matplotlib.Axes .Есть ли какой-нибудь возможный способ сделать это?

Я прикрепил код ниже:

current_ax = np.empty((3,3), dtype = object)

df = pd.read_csv('data.csv')

data = []

for i in range(3):
    for j in range(3):
        current_ax[i,j] = fig.add_subplot(current_gs[i,j],label=f"{i}{j}")

        if i == 2:
           current_ax[i,j].set_ylabel(str(j))
        if j == 0:
            current_ax[i,j].set_xlabel(str(i))

        query = "y==" + str(j) + " and y_pred==" + str(i)
        col = 'y_proba' + str(i)
        results = df.query(query)
        data.append(results[col].item())

# Now I want to convert current_ax to an Axes object

Спасибо, что посмотрели.