在熊猫数据集上执行SQL查询

Question

我有一个名为“ df”的熊猫数据集。

我该如何做以下事情？

df.query("select * from df")

谢谢。

对于那些了解R的人，有一个名为sqldf的库，您可以在R中执行SQL代码，我的问题基本上是，是否在python中有像sqldf这样的库

Answer 1

这不pandas.query应该做，您可以看一下包pandasql（与sqldfR中的一样）

import pandas as pd
import pandasql as ps

df = pd.DataFrame([[1234, 'Customer A', '123 Street', np.nan],
               [1234, 'Customer A', np.nan, '333 Street'],
               [1233, 'Customer B', '444 Street', '333 Street'],
              [1233, 'Customer B', '444 Street', '666 Street']], columns=
['ID', 'Customer', 'Billing Address', 'Shipping Address'])

q1 = """SELECT ID FROM df """

print(ps.sqldf(q1, locals()))

     ID
0  1234
1  1234
2  1233
3  1233

更新2020-07-10

更新 pandasql

ps.sqldf("select * from df")