Problem extracting words from dataframe
The problem is that you say &quot;apply(lambda x:[i.split()[0] for i in x])&quot;
As soon as you say apply, x is your list. So you can say following
&quot;apply(lambda x:x[0].split(&quot; &quot;, 2)[0])&quot;
Meaning you say take first element in the list, than split on &quot; &quot; and in two parts. And than take the first word (part) with the last [0]