Разбить столбец в пандах на основе ряда строк

Question

Я пытаюсь разбить столбец с текстом в пандах.

В этой колонке содержатся слова «Альфа», «Бета». Мне нужно, чтобы этот столбец разделился после этих слов, этих слов и всего, прежде чем нужно было идти в один столбец и все после этого в новом столбце.

Я попытался это:

split_df = df.str.split('ALPHA ').apply(pd.Series) 

Но тогда я потеряю все, прежде чем ALPHA

Answer 1

IIUC вы можете split по регулярному выражению r'Alpha|Beta' - это означает Alpha или Beta:

print df 
      text 
0 aaa Alpha bbb 
1 ddd Alpha ccc 
2 uuu Beta vvv 

split_df = df['text'].str.split(r'Alpha|Beta').apply(pd.Series) 
split_df.columns = ['before','after'] 
print split_df 
    before after 
0 aaa  bbb 
1 ddd  ccc 
2 uuu  vvv