我想从整个数据库中获取列名(假设数据库包含超过100行,其中50列以上)基于熊猫中特定列中包含的特定值。
在Bkmm3(来自印度的成员)的帮助下,我在数字术语上取得了成功,但在字母术语上却失败了。我尝试的方式是这样的:
df = pd.DataFrame({'A':['APPLE','BALL','CAT'],
'B':['ACTION','BATMAN','CATCHUP'],
'C':['ADVERTISE','BEAST','CARTOON']})
response = input("input")
for i in df.columns: if(len(df.query(i + '==' + str(response))) > 0):
print(i)`
Run Code Online (Sandbox Code Playgroud)
然后输出出现错误:
Traceback (most recent call last): NameError: name 'APPLE' is not defined
Run Code Online (Sandbox Code Playgroud)
谢谢大家的任何帮助,谢谢。。。