【发布时间】:2015-04-22 14:23:13
【问题描述】:
我有一个带有多索引的数据框,我想根据一些基于索引的模式从中删除行。例如,我想在下面的数据框中删除注释器为“Peter Test xx”且标签为“空”的第 1-4 帧
print df
boundingbox x1 boundingbox y1 \
frame annotator label
0 Peter Test xx empty NaN NaN
1 Peter Test xx empty NaN NaN
2 Peter Test xx empty NaN NaN
3 Peter Test xx empty NaN NaN
Petaa yea NaN NaN
4 Peter Test xx empty NaN NaN
5 P empty frame 494 64
Peter Test xx empty NaN NaN
6 P empty frame 494 64
Peter Test xx empty NaN NaN
7 P empty frame 494 64
Peter Test xx empty NaN NaN
8 P empty frame 494 64
Peter Test xx empty NaN NaN
我可以通过执行类似的操作来选择行
indexer = [slice(None)]*len(df.index.names)
indexer[df.index.names.index('frame')] = range(1,4)
indexer[df.index.names.index('annotator')] = ['Peter Test xx']
indexer[df.index.names.index('label')] = ['empty']
return df.loc[tuple(indexer),:]
如果我想删除这些行,理想情况下我想做类似的事情
del df.loc[tuple(indexer),:]
但这不起作用(为什么?)。我在网上找到的所有解决方案都是基于 int 的索引。但是如果我使用字符串作为索引,我不能简单地切片或类似的东西。
我也尝试过:
def filterFunc(x, frames, annotator, label):
if x[0] in frames\
and x[1] == annotator\
and x[2] == label:
return 1
else:
return 0
mask = df.index.map(lambda x: filterFunc(x, frames, annotator, label))
return df[~mask,:]
这给了我:
TypeError: unhashable type: 'numpy.ndarray'
有什么建议吗?
【问题讨论】:
标签: python-2.7 pandas