Pandas dataframe：使用条件和切片字符串创建新列

Question

我有这段代码创建一个新的 dataframe 列，首先使用条件，然后使用固定的切片索引（0、5）切片一些字符串：

df.loc[df['operation'] == 'dividend', ['order_adj']] = df['comment'].str.slice(0, 5)

但是，我需要在这段代码的最后使用str.find()来根据其字符在df['comment']上拥有一个动态切片索引，而不是使用固定的切片索引。

当我通过广播创建一个新列时，我找不到正确的 sintaxe 以inside str.slice()中使用str.find('some_string') ) 。 谢谢。

Answer 1

使用split的选项：

df['comment'].str.split("some_string").str[0]

或使用正则表达式的选项（将捕获组移动到您想要的包含/独占的位置）：

pandas.Series.str.extract("(.*?)some_string")
pandas.Series.str.extract("(.*?some_string)")