It may be preferable to create a variable tied to a UI element that can be used as a toggle to view each analysis. choice = mo.ui.dropdown(['train', 'split')

It may be preferable to create a variable tied to a UI element that can be used as a toggle to view each analysis. choice = mo.ui.dropdown(['train', 'split') data = load(choice.value) processed = process(data) visualize(processed) This way, you can toggle between just more than two if needed. If you need to see both at once, you'd want to refactor the processing and visualizing step into functions, and then just duplicate the finals cell(s). marimo has a multi-column mode, so you can view them side-by-side