给我一本字典
dictionary = {'Animal 1': {'Dog': 'Yes', 'Cat': 'No', 'Color': 'Black'},
'Animal 2': {'Dog': 'Yes', 'Cat': 'No', 'Color': 'Brown'},
'Animal 3': {'Dog': 'No', 'Cat': 'Yes', 'Color': 'Grey'}}
如何选择狗类动物?预期输出['Animal 1','Animal 2']
我可以用途:
pd.DataFrame.from_dict(dictionary).T.loc[pd.DataFrame.from_dict(dictionary).T["Dog"]=='Yes',:].index.to_list()
但是看起来很丑
2条答案
按热度按时间7cjasjjr1#
您可以使用列表解析:
印刷品:
mi7gmzs62#
pandas版本可以通过使用一个中间变量进行微调,这样你就不会重复计算掩码,而且你不需要为这个过滤器使用.loc。
但是,与在这个问题的另一个答案中运行指令项相比,这仍然是复杂的。