I'm trying to filter a Pandas dataframe using a string and function query()
on a timestamp column:
df.query('Timestamp < "2020-02-01"')
However, I get the following error:
Traceback (most recent call last):
File "C:\ENERCON\Python 3.7.2\lib\site-packages\IPython\core\interactiveshell.py", line 3326, in run_code
exec(code_obj, self.user_global_ns, self.user_ns)
File "<ipython-input-3-7bb40e9c631a>", line 1, in <module>
df.query('Timestamp < "2020-02-01"')
File "C:\ENERCON\Python 3.7.2\lib\site-packages\pandas\core\frame.py", line 3199, in query
res = self.eval(expr, **kwargs)
File "C:\ENERCON\Python 3.7.2\lib\site-packages\pandas\core\frame.py", line 3315, in eval
return _eval(expr, inplace=inplace, **kwargs)
File "C:\ENERCON\Python 3.7.2\lib\site-packages\pandas\core\computation\eval.py", line 327, in eval
ret = eng_inst.evaluate()
File "C:\ENERCON\Python 3.7.2\lib\site-packages\pandas\core\computation\engines.py", line 142, in evaluate
return self.expr()
File "C:\ENERCON\Python 3.7.2\lib\site-packages\pandas\core\computation\expr.py", line 837, in __call__
return self.terms(self.env)
File "C:\ENERCON\Python 3.7.2\lib\site-packages\pandas\core\computation\ops.py", line 380, in __call__
return self.func(left, right)
TypeError: '<' not supported between instances of 'type' and 'str'
Also tried to convert the string to datetime, but the error is similar.
df.query('Timestamp < @pd.to_datetime("2020-02-01")')
Traceback (most recent call last):
File "C:\ENERCON\Python 3.7.2\lib\site-packages\IPython\core\interactiveshell.py", line 3326, in run_code
exec(code_obj, self.user_global_ns, self.user_ns)
File "<ipython-input-5-23540526aad9>", line 1, in <module>
df.query('Timestamp < @pd.to_datetime("2020-02-01")')
File "C:\ENERCON\Python 3.7.2\lib\site-packages\pandas\core\frame.py", line 3199, in query
res = self.eval(expr, **kwargs)
File "C:\ENERCON\Python 3.7.2\lib\site-packages\pandas\core\frame.py", line 3315, in eval
return _eval(expr, inplace=inplace, **kwargs)
File "C:\ENERCON\Python 3.7.2\lib\site-packages\pandas\core\computation\eval.py", line 322, in eval
parsed_expr = Expr(expr, engine=engine, parser=parser, env=env, truediv=truediv)
File "C:\ENERCON\Python 3.7.2\lib\site-packages\pandas\core\computation\expr.py", line 830, in __init__
self.terms = self.parse()
File "C:\ENERCON\Python 3.7.2\lib\site-packages\pandas\core\computation\expr.py", line 847, in parse
return self._visitor.visit(self.expr)
File "C:\ENERCON\Python 3.7.2\lib\site-packages\pandas\core\computation\expr.py", line 441, in visit
return visitor(node, **kwargs)
File "C:\ENERCON\Python 3.7.2\lib\site-packages\pandas\core\computation\expr.py", line 447, in visit_Module
return self.visit(expr, **kwargs)
File "C:\ENERCON\Python 3.7.2\lib\site-packages\pandas\core\computation\expr.py", line 441, in visit
return visitor(node, **kwargs)
File "C:\ENERCON\Python 3.7.2\lib\site-packages\pandas\core\computation\expr.py", line 450, in visit_Expr
return self.visit(node.value, **kwargs)
File "C:\ENERCON\Python 3.7.2\lib\site-packages\pandas\core\computation\expr.py", line 441, in visit
return visitor(node, **kwargs)
File "C:\ENERCON\Python 3.7.2\lib\site-packages\pandas\core\computation\expr.py", line 747, in visit_Compare
return self.visit(binop)
File "C:\ENERCON\Python 3.7.2\lib\site-packages\pandas\core\computation\expr.py", line 441, in visit
return visitor(node, **kwargs)
File "C:\ENERCON\Python 3.7.2\lib\site-packages\pandas\core\computation\expr.py", line 565, in visit_BinOp
return self._maybe_evaluate_binop(op, op_class, left, right)
File "C:\ENERCON\Python 3.7.2\lib\site-packages\pandas\core\computation\expr.py", line 547, in _maybe_evaluate_binop
return self._maybe_eval(res, self.binary_ops)
File "C:\ENERCON\Python 3.7.2\lib\site-packages\pandas\core\computation\expr.py", line 519, in _maybe_eval
self.env, self.engine, self.parser, self.term_type, eval_in_python
File "C:\ENERCON\Python 3.7.2\lib\site-packages\pandas\core\computation\ops.py", line 399, in evaluate
res = self(env)
File "C:\ENERCON\Python 3.7.2\lib\site-packages\pandas\core\computation\ops.py", line 380, in __call__
return self.func(left, right)
TypeError: '<' not supported between instances of 'type' and 'Timestamp'
If I run the equivalent function using .loc
I have the desired results (but I can't use a user input string).
df.loc[df['Timestamp'] < "2020-02-01"]
Out[4]:
Timestamp Error ... ToD Day_Night
0 2020-01-17 00:00:00 0 ... 0 Night
1 2020-01-17 00:10:00 0 ... 0 Night
2 2020-01-17 00:20:00 0 ... 0 Night
3 2020-01-17 00:30:00 0 ... 0 Night
4 2020-01-17 00:40:00 0 ... 0 Night
2154 2020-01-31 23:10:00 0 ... 23 Night
2155 2020-01-31 23:20:00 0 ... 23 Night
2156 2020-01-31 23:30:00 0 ... 23 Night
2157 2020-01-31 23:40:00 0 ... 23 Night
2158 2020-01-31 23:50:00 0 ... 23 Night
[2159 rows x 37 columns]
Does anyone knows how to use query()
with datetime columns?
The Timestamp
column name shadows the built-in type timestamp
. As a first step, you can rename the column to something else, using rename()
:
df.rename(columns={"Timestamp": "MyTimestamp"})
Then the following should do the trick for the datetime:
df.query('MyTimestamp < 20200201')
Alternatively, if you want to query the dataframe using a timestamp:
df.query('MyTimestamp < @ts("20200201T071320")'
The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.