Returning optimized light subcomponents from model plot

[amn3142]

What would the best way be to get the subcomponents from model plot as arrays? I see how to get them as separate plots, but I would like to just have the arrays myself.

There's probably a way to extract the data from the image figure, but I think adding in an option to return array might also be useful in model plot. Does such an option already exist?

Motivation: I want to make a lot of modifications to how I am plotting including things like overall normalization and fov that I think will be easier for me to handle in a uniform way on my own.

Thanks in advance.

[sibirrer]

Hi @amn3142,
this class might be of interest to you:

lenstronomy/lenstronomy/Analysis/image_reconstruction.py

Line 12 in 4c5e6dc

class MultiBandImageReconstruction(object):

You can generate then individual ImageModel classes with all the solution in there and then you can execute individual components.

[amn3142]

It was just a couple of lines of code to get the decomposition plot in model plot to return arrays rather than a figure when a new 'return_array' boolean was set.

It is hacky because I still create and pass it an axis object that goes unused, but it works without me having to think so that's a major plus. I can submit a pull request if you think it would be a useful feature for others.

[sibirrer]

Hi @amn3142, great to hear it worked! You can still make a PR, that could help make the design be improved for 'less of a hack' perhaps in the future, and also to make sure your feature is forward compatible. Thank you!