Создание нового файла изображения в астропии с указанными размерами другого файла изображения

Я думаю, что ответ @VincePs правильный, но я добавлю дополнительную информацию, потому что я думаю, что вы не используете здесь возможности astropy.

Во-первых, Python начинается с нуля, поэтому основное расширение имеет номер 0. Может быть, вы ошиблись, а может, и нет - но получить доступ ко второму HDU редко, поэтому я подумал, что лучше упомянуть об этом.

hdu_num = 0 # Or use = 1 if you really want the second hdu.

Во-первых, вам не нужно открывать один и тот же файл дважды, вы можете открыть его один раз и закрыть после извлечения соответствующих значений:

with fits.open('...') as hdus:
    nxpix_um2_ext1 = hdus[hdu_num]['NAXIS1']
    nxpix_um2_ext1 = hdus[hdu_num]['NAXIS2']
# Continue without indentation and the file will be closed again.

или если вы хотите сохранить весь заголовок (для его сохранения позже) и данные, которые вы можете использовать:

with fits.open('...') as hdus:
    hdr = hdus[hdu_num].header
    data = hdus[hdu_num].data # I'll also take the data for comparison.

Я продолжу использовать второй подход, потому что я думаю, что он намного чище, и у вас будут готовы все данные и значения заголовков.

new_data = np.zeros((hdr['NAXIS2'], hdr['NAXIS1']))

Обратите внимание, что Python интерпретирует ось иначе, чем IRAF (и я думаю, что IDL, но я не уверен), поэтому вам понадобится axis2 как первый и axis1 как второй элемент.

Так что быстро проверьте, совпадают ли формы:

print(new_data.shape)
print(data.shape)

Если они не равны, я запутался в оси в Python (снова), но я так не думаю. Но вместо создания нового массива на основе значений header вы также можете создать новый массив, просто используя старую форму:

new_data_2 = np.zeros(data.shape)

Это обеспечит идентичность размеров и формы. Теперь у вас есть пустое изображение. Если вам нравится копия, вы можете, но не обязаны, явно копировать данные (кроме случаев, когда вы открывали файл явно в режиме записи / добавления / обновления, вы всегда должны копировать его, но это не по умолчанию).

new_data = data # or = data.copy() for explicitly copying.

Выполните свои операции с ним, и если вы хотите сохранить его снова, вы можете использовать то, что предложил @VinceP:

hdu = fits.PrimaryHDU(new_data, header=hdr) # This ensures the same header is written to the new file
hdulist = fits.HDUList([hdu])
hdulist.writeto('new.fits')

Обратите внимание, что вам не нужно изменять ключевые слова заголовка, связанные с формой, даже если вы изменили форму данных, потому что во время writeto астропия обновит их (по умолчанию)

В документации астропии вы выполните эту задачу шаг за шагом: создайте массив с size (NAXIS1, NAXIS2), поместите данные в первичный HDU, создайте HDUlist и запишите его на диск:

import numpy as np
from astropy.io import fits

data = np.zeros((NAXIS2,NAXIS1))
hdu = fits.PrimaryHDU(data)
hdulist = fits.HDUList([hdu])

hdulist.writeto('new.fits')