python - 查找 Pandas Series 包含包含字符的元素的位置的索引

Question

例子:

import pandas as pd    
arr = pd.Series(['a',['a','b'],'c'])

我想获取系列包含包含 'a' 的元素的位置的索引。所以我想取回索引 0 和 1。

我试过写

arr.str.contains('a')

但这会返回

0     True
1      NaN
2    False
dtype: object

虽然我希望它返回

0     True
1     True
2    False
dtype: object

Answer 1

使用Series.str.join()将单元格中的列表/数组连接成一个字符串，然后使用 .str.contains('a'):

In [78]: arr.str.join(sep='~').str.contains('a')
Out[78]:
0     True
1     True
2    False
dtype: bool