Преобразовать часть списка в Pandas Dataframe

Question

У меня есть список 'list_of_categories_train' со значениями:

0                                      [Men, Tops, T-shirts]
1          [Electronics, Computers & Tablets, Components ...
2                            [Women, Tops & Blouses, Blouse]
3                   [Home, Home Décor, Home Décor Accents]
4                                [Women, Jewelry, Necklaces]
                                 ...                        
1048570                                 [Kids, Other, Other]
1048571           [Other, Daily & Travel items, Health Care]
1048572    [Women, Athletic Apparel, Pants, Tights, Leggi...
1048573                       [Men, Shoes, Fashion Sneakers]
1048574                    [Kids, Toys, Dolls & Accessories]
Name: category_name, Length: 1048575, dtype: object

Мне нужно выбрать 1-е слово каждого списка и поместить его в фрейм данных. Но это занимает часы. Какой самый быстрый способ сделать это? Мой код:

i=0
while i <= len(list_of_categories_train)  :  
    data_train['noOfCategories']= list_of_categories_train[i][0]
    i=i+1

Answer 1

Предполагая, что list_of_categories_train является вашей серией, , и если каждый элемент серии является списком , вы можете попробовать:

data_train['noOfCategories'] = list_of_categories_train.apply(lambda x: x[0])

Answer 2

Использование:

data_train['noOfCategories'] = [x[0] for x in list_of_categories_train]

Медленнее:

data_train['noOfCategories'] = list_of_categories_train.str[0]