From a function in a jupyter notebook to a dataframe

Hi Team,

I have created a simple recommendation system in a python notebook. I started by changing the dataiku.dataset to a dictionary.

This is the final function:

Here you can see the results for family nr. 13.

However, I would like to get recommendations for all family-numbers. The output should be five columns. The first column holding the top recommendation for each family, the second column holding the next best recommendation, etc