我有一个包含 3 列的数据框,我需要获取与 2 列的值匹配的行的索引。
例如,下面的数据框:
Name City Country
Pietro Roma Italy
Maria Milan Italy
Pietro NY USA
在这种情况下,我需要获取 Pietro|Roma|Italy 的索引,仅搜索名称和城市列。
我尝试执行下面的代码,但它返回了与 2 列匹配的所有行。
idx = np.where(dataframe[dataframe["Name"] == 'Pietro'],dataframe[dataframe["City"] == 'Roma'])
但它返回一个索引数组 [[0,2],[0]],我需要返回索引 0,即 Name = 'Pietro' 和 City = 'Roma'
更新了解决方案
解决方法是:
dataframe.index[(dataframe["Name"] == 'Pietro')&(dataframe["City"] == 'Roma')][0]
最佳答案
使用
dataframe.index[(dataframe["Name"] == 'Pietro')&(dataframe["City"] == 'Roma')]
关于python - 查找匹配 2 列条件的单个数据帧行索引,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/54873393/