@simonrayner first question would be if you have given KNIME enough RAM, and then there are several more things to consider:
And in the end, there is a seamless integration between KNIME and Python (data transfer via Arrow). So, if you must you can do data preparations in Python and then use KNIME 