KeyError: 0 # If we have a listlike key, _check_indexing_error will raise after applying a function to a pandas dataframe

I have a function f(featureList) that takes a list of arbitrary length as input and gives another list of the same length as output

Pandas dataframe groupby apply function with variable number of arguments

I have a pandas dataframe that looks like