Построение сетки WCS на срезе из файла 3-мерных подгонок в python
Я работаю с 3-мерным файлом подгонки с двумя пространственными измерениями, а третьим является длина волны. Я пытаюсь построить срез с координатами положения по осям X и Y, что мне удалось использовать следующий код:
hdu =fits.open('filename')[0]
wcs = WCS(hdu.header)
image_data = hdu.data
fig = plt.figure()
ax = plt.subplot(projection=wcs, slices=('x', 'y', 1083))
plt.imshow(image_data[1083, :, :].transpose())
plt.show()
что дает мне это изображение введите описание изображения здесь
однако мне бы очень хотелось, чтобы координаты wcs были наложены на изображение, а не отмечены галочками. Я пробовал добавлять
ax.coords.grid(color='white', alpha=0.5, linestyle='solid')
но это дает мне следующую ошибку:
... /. local / lib / python2.7 / sitepackages / astropy / visualization / wcsaxes / transforms.py ", строка 98, в повышении transform ValueError(" Второе измерение входных значений должно соответствовать числу координат WCS ") ValueError: Второе измерение входных значений должно соответствовать количеству координат WCS
Я подозреваю, что эта ошибка вызвана тем, что файл соответствий был трехмерным, но я не смог решить / исправить ошибку. у кого-нибудь есть решение?