我正在尝试在 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 ,我不清楚如何使用索引来提高上述查询的性能。有没有办法可以使用索引提高上述查询的性能?
最佳答案
您可以查询底层 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
如果要保留原始索引,请在运行查询之前将其转换为列,然后再返回索引。
关于python - Pandas MultiIndex 的简单用例,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/49089238/