創建與條件符合的行相對應的單獨索引

Question

我敢打賭，對於我的問題有一個簡單的答案，但我正在努力根據熊貓文檔找出答案。

我有一個大型數據框，約20,000行14列。我需要找到其中某行中包含某個正則表達式表達式的索引。因此，對於一個非常簡單的例子：

label1 label2 
0 cat  dog 
1 bird  monkey 
2 elephant bobcat 

當我搜索「貓」，我想獲得一個系列，列表，某種迭代的，看起來像

0 
2 

你將如何用熊貓做那個？對於20,000或更高的數據集，我可以每秒做2-3次，所以速度在這裏也非常重要。

Answer 1

不知道它可以在您的數據集上執行多快。

df.applymap(lambda x: 'cat' in x).sum(1).nonzero() 
Out[12]: (array([0, 2], dtype=int64),)