Trying to get a hold of pd.query function
Getting an error
UndefinedVariableError: name ‘BACKTICK_QUOTED_STRING__AT_key’ is not defined
for the below pandas Python code. What did I do wrong?
JavaScript
x
13
13
1
def ft_equal(df,key,value):
2
# print(key)
3
# print(value)
4
# return df[df[key] == value]
5
# print(key)
6
# print(df[key])
7
# return df.query('@key == "%s"' %value)
8
return df.query('`@key` == @value')
9
10
pd.DataFrame.ft_equal = ft_equal
11
12
df.ft_equal("Current State","Disburse")
13
Advertisement
Answer
Try this :
JavaScript
1
4
1
def ft_equal(df,key,value):
2
my_query = key + " == " + value
3
return df.query(my_query)
4
or
JavaScript
1
3
1
def ft_equal(df,key,value):
2
return df.query(f"{key} == {value}")
3