0

I have a function preprocess_names that splits an input variable:

def preprocess_names(name):
    #SOME PREPROCESSING
    return name.split('')

Also, I have a Pandas DataFrame df to which I want to apply this function. In my case, the preprocess_names's output has length 3 and I want to use each value in a new column in df.

So, for the first new column, I want to use the first element from the output list. If I used this function on one str only, I would use preprocess_names[0] to get the correct variable. How can I do something similar while applying the method to df?

The code below raises ValueError, but is closest to what I would like to do.

df['runs'] = df['name'].apply(preprocess_names)[0]
Emil
  • 1,531
  • 3
  • 22
  • 47

1 Answers1

1

Do you want:

df['runs'] = df['name'].apply(preprocess_names).str[0]
U13-Forward
  • 69,221
  • 14
  • 89
  • 114