python - Pandas 有效检查列是否包含其他列中的字符串

Question

我正在尝试获取一个 bool 值索引，以确定一列是否包含来自另一列中同一行的字符串:

a      b
boop   beep bop
zorp   zorpfoo
zip    foo zip fa

检查 b 列是否包含字符串，我想得到:

[False, True, True]

现在我正在尝试这种方法，但它很慢:

df.apply(lambda row: row['a'] in row['b'], axis=1)

是否有 .str 方法？

Answer 1

df.apply(..., axis=1) 非常慢!你应该避免使用它!

from random import sample
from string import lowercase
from pandas import DataFrame

df = DataFrame({
    'a': map(lambda x: ''.join(sample(lowercase, 2)), range(100000)),
    'b': map(lambda x: ''.join(sample(lowercase, 5)), range(100000))
})

%time map(lambda (x, y): x in y, zip(df['a'], df['b']))

%time df.apply(lambda x: x[0] in x[1], axis=1)