Pandas MultiIndex的简单用例

Question

我正在尝试在Pandas数据帧上编写一个简单的查询

A B C
0 1 2.1
0 2 3.0
0 3 4.0
1 0 4.0

我想选择所有行WHERE A=0 and B>1所以我期望作为输出

A B C
0 2 3.0
0 3 4.0

我定义我的DataFrame如下

df = pd.DataFrame([{'A': 0, 'B': 1, 'C': 2.1}, {'A': 0, 'B': 2, 'C': 3.0}, {'A': 0, 'B': 3, 'C': 4.0}, {'A': 1, 'B': 0, 'C': 4.0}])

然后执行以下查询

df[ (df["A"] == 0) & (df["B"] > 1) ]

   A  B    C
1  0  2  3.0
2  0  3  4.0

它有效但在我的实际用例中它很慢，因为还有更多的行。

经过pandas MultiIndexing doc之后，我不清楚我如何使用索引来改善上述查询的性能。有没有办法可以使用索引提高上述查询的性能？

Answer 1

您可以查询基础numpy数组而不是原始数据帧：

%timeit df[(df["A"] == 0) & (df["B"] > 1) ]
#1000 loops, best of 3: 1.23 ms per loop

ar = df.values
%timeit ar[(ar[:,0] == 0) & (ar[:,1] > 1)]
#100000 loops, best of 3: 11.5 µs per loop

如果要保留原始索引，请在运行查询之前将其转换为列，然后再返回索引。