Let’s break down why you’re hitting this error and fix your code step by step.

Root Cause of the Error #

The main issue is that you’re assigning the entire HDUList object (returned by fits.open()) to coeff, instead of extracting the actual numeric data array from the FITS file. When you try to index into coeff[0,i,j], Python tries to unpack the file object like a regular array—which it isn’t—triggering the "too many values to unpack" error.

Step-by-Step Fixes #

1. Correctly Extract Coefficient Data #

First, you need to pull the actual data array from your coefficient FITS file. Assuming your coefficients are stored in the first HDU (index 0), update this part of your code. Also, use a context manager (with statement) to automatically close the file when you’re done, avoiding resource leaks:

with fits.open('MCtest_C1.fits') as hdu1:
    coeff = hdu1[0].data  # Extract the numeric data array from the HDU

2. Verify Coefficient Array Shape #

Your correction formula expects coeff to be a 3D array with shape (4, 1024, 1024) (4 coefficients for each 1024x1024 pixel). Check the shape with:

print(coeff.shape)

If the shape is (1024, 1024, 4) instead, transpose it to get coefficients in the first dimension:

coeff = coeff.transpose(2, 0, 1)

3. Ditch Nested Loops for Numpy Broadcasting #

Iterating over every pixel with nested loops is slow and error-prone. Numpy’s broadcasting lets you compute the correction for the entire image in one line:

# Calculate non-linear correction for all pixels at once
nonlin = coeff[0] * img**3 + coeff[1] * img**2 + coeff[2] * img + coeff[3]

Note: img**0 simplifies to 1, so we can use coeff[3] directly instead of multiplying by img**0.

4. Fix the Undefined filelist Variable #

The line corrected_img = (filelist)/nonlin uses filelist, which isn’t defined in your code. Assuming this is supposed to be your original image, replace filelist with img (or the correct input array):

corrected_img = img / nonlin

5. Clean Up Open FITS Files #

Use a context manager for your original image file too, to ensure it gets closed properly:

with fits.open('1/img/r1464838.fit') as hdu:
    img = hdu[1].data
print(img.shape)

Corrected Full Code #

from astropy.io import fits
import matplotlib.pyplot as plt
import numpy as np

# Read original image safely with a context manager
with fits.open('1/img/r1464838.fit') as hdu:
    img = hdu[1].data
print(f"Original image shape: {img.shape}")

# Read coefficient data
with fits.open('MCtest_C1.fits') as hdu1:
    coeff = hdu1[0].data
print(f"Coefficient shape (before adjustment): {coeff.shape}")

# Ensure coefficients match the expected (4, 1024, 1024) shape
if coeff.shape != (4, 1024, 1024):
    coeff = coeff.transpose(2, 0, 1)
    print(f"Coefficient shape (after adjustment): {coeff.shape}")

# Compute non-linear correction efficiently
nonlin = coeff[0] * img**3 + coeff[1] * img**2 + coeff[2] * img + coeff[3]

# Calculate corrected image
corrected_img = img / nonlin

# Optional: Visualize results
plt.figure(figsize=(12, 6))
plt.subplot(121)
plt.imshow(img, cmap='gray')
plt.title('Original Image')
plt.colorbar()

plt.subplot(122)
plt.imshow(corrected_img, cmap='gray')
plt.title('Corrected Image')
plt.colorbar()

plt.tight_layout()
plt.show()

Additional Tips #

• Always check array shapes with print(arr.shape) when debugging array operations—mismatched shapes are a common source of errors.
• Context managers are the safest way to handle FITS files, as they automatically close files even if an error occurs.

内容的提问来源于stack exchange，提问作者helena