Как извлечь слово из строк в Pandas DataFrame

Question

Если у меня есть категория имени столбца, и у меня есть такие строки, как Путешествие на самолете | Путешествие на поезде | Путешествие на автобусе, как я могу извлечь Путешествие на самолете в pandas Dataframe

Answer 1

Вы можете напрямую извлечь первое значение, используя .str в пандах.

# sample data frame
df = pd.DataFrame({'category': ['Plane Travel|Train Travel|Bus Travel ','Plane Travel|Train Travel|Bus Travel ','Plane Travel|Train Travel|Bus Travel ']})

# new column
df['new_col'] = df['category'].str.split('|').str[0]

print(df)

                               catgeory       new_col
0  Plane Travel|Train Travel|Bus Travel   Plane Travel
1  Plane Travel|Train Travel|Bus Travel   Plane Travel
2  Plane Travel|Train Travel|Bus Travel   Plane Travel

Answer 2

Вам нужно использовать аксессор .str и затем .split() вашу строку, тогда вы можете поместить результат в отдельные столбцы.

Давайте сгенерируем правильный DataFrame:

df = pd.DataFrame({"Category":["Plane France", "Train Russia", "Spacecraft Moon"],
                   "other_variable":[1,2,3] })
print df

       Category  other_variable
0  Plane France               1
1  Train Russia               2
2  Spacecraft Moon            3

Теперь вы можете получить доступ к строкам с помощью .str (взгляните на Pandas doc ) и разделить их.

df["category_list"] = df.Category.str.split(" ") # you can replace " " with any   
                                                 # other word delimiter

и затем вы должны присвоить каждому элементу списка новый столбец

df[["transportation", "destination"]] = pd.DataFrame(df.category_list.values.tolist(), 
                                                     index = df.index)

что дает

          Category  other_variable       category_list transportation  \
0     Plane France               1     [Plane, France]          Plane   
1     Train Russia               2     [Train, Russia]          Train   
2  Spacecraft Moon               3  [Spacecraft, Moon]     Spacecraft   

  destination  
0      France  
1      Russia  
2        Moon  

Теперь у вас есть столбцы назначения для транспортировки.