python - 使用 Pandas 在另一列中查找列字符串的位置

Question

我有一个包含 2 列的 Dataframe

   col1     col2
1  cat      the cat
2  dog      a nice dog
3  horse    horse is here

我需要找到 col1 的每个字符串在 col2 中的位置。

解决方案必须是:

   col1     col2          col3
1  cat      the cat        4
2  dog      a nice dog     7
3  horse    horse is here  0

必须有一个简单的解决方案来执行此操作而不使用痛苦的循环，但我找不到它。

Answer 1

numpy.core.defchararray.find

from numpy.core.defchararray import find

a = df.col2.values.astype(str)
b = df.col1.values.astype(str)
df.assign(col3=find(a, b))

    col1           col2  col3
1    cat        the cat     4
2    dog     a nice dog     7
3  horse  horse is here     0