Is there a Python function for removing variables from a list?

Question

I'm looking for a way to clean up this code. I'm using a curve_fit function to spit out fitted coefficients for various functions which have different numbers of parameters. I then want to use these fitted parameters to predict the outputs. The problem is, I'm using these clunky if/else statements to check how many parameters I need to feed into my function. I feel like there is an obvious way to achieve this much more cleanly but I just can't seem to see it.

fit = op.curve_fit(func, ordered_radii, ordered_masses, p0 = p0, bounds = bounds, maxfev = 10000000)[0]

if len(fit) == 1:
    predicted_masses = func(ordered_radii, fit[0])
    
elif len(fit) == 2:
    predicted_masses = func(sieve_radii, fit[0], fit[1])
    
elif len(fit) == 3:
    predicted_masses = func(ordered_radii, fit[0], fit[1], fit[2])
        
elif len(fit) == 4:
    predicted_masses = func(ordered_radii, fit[0], fit[1], fit[2], fit[3])

Thank you very much.

question from:https://stackoverflow.com/questions/66047931/is-there-a-python-function-for-removing-variables-from-a-list

Answer 1

Waitting for answers