这明显是一个bug。
s1.index.tolist()返回的值不包含"bar"。
>>> s1.index.tolist()
[('baz', 'one'), ('baz', 'two'), ('foo', 'one'), ('foo', 'two')]
s1["bar"] 返回一个空的Series。
>>> s1["bar"]
Series([], dtype: float64)
标准的覆盖方法似乎也不起作用:
>>> del s1["bar"]
>>> s1["bar"]
Series([], dtype: float64)
>>> s1.__delitem__("bar")
>>> s1["bar"]
Series([], dtype: float64)
然而,正如预料的那样,尝试获取一个新的键会引发KeyError错误:
>>> s1["booz"]
... KeyError: 'booz'
主要区别在于当你实际查看pandas.core.index.py中的源代码时。
class MultiIndex(Index):
...
def _get_levels(self):
return self._levels
...
def _get_labels(self):
return self._labels
def tolist(self):
"""
return a list of the Index values
"""
return list(self.values)
因此,index.tolist()和_labels没有访问相同的共享信息,事实上,它们甚至不接近。
因此,我们可以使用这个方法手动更新结果索引器。
>>> s1.index.labels
FrozenList([[1, 1, 2, 2], [0, 1, 0, 1]])
>>> s1.index._levels
FrozenList([[u'bar', u'baz', u'foo'], [u'one', u'two']])
>>> s1.index.values
array([('baz', 'one'), ('baz', 'two'), ('foo', 'one'), ('foo', 'two')], dtype=object)
如果我们将此与最初的多级索引进行比较,我们会得到以下结果
>>> s.index.labels
FrozenList([[0, 0, 1, 1, 2, 2], [0, 1, 0, 1, 0, 1]])
>>> s.index._levels
FrozenList([[u'bar', u'baz', u'foo'], [u'one', u'two']])
所以_levels属性没有更新,而值已经更新。
编辑:覆盖它并不像我想的那么简单。
编辑:编写自定义函数来修复此行为。
from pandas.core.base import FrozenList, FrozenNDArray
def drop(series, level, index_name):
new_series = series.drop(index_name)
levels = new_series.index.levels
labels = new_series.index.labels
index_pos = levels[level].tolist().index(index_name)
level_names = levels[level]
tmp_names = FrozenList([i for i in level_names if i != index_name])
levels = FrozenList([j if i != level else tmp_names
for i, j in enumerate(levels)])
new_series.index.set_levels(levels, verify_integrity=False, inplace=True)
level_labels = labels[level].tolist()
tmp_labels = FrozenNDArray([i-1 if i > index_pos else i
for i in level_labels])
labels = FrozenList([j if i != level else tmp_labels
for i, j in enumerate(labels)])
new_series.index.set_labels(labels, verify_integrity=False, inplace=True)
return new_series
示例用户:
>>> s1 = drop(s, 0, "bar")
>>> s1.index
MultiIndex(levels=[[u'baz', u'foo'], [u'one', u'two']],
labels=[[0, 0, 1, 1], [0, 1, 0, 1]])
>>> s1.index.tolist()
[('baz', 'one'), ('baz', 'two'), ('foo', 'one'), ('foo', 'two')]
>>> s1["bar"]
...
KeyError: 'bar'
编辑:这似乎只适用于具有多索引的数据帧/系列,因为标准的pandas.core.index.Index类没有相同的限制。我建议提交错误报告。
考虑具有标准索引的相同系列:
>>> s = p.Series(np.random.randn(6))
>>> s.index
Int64Index([0, 1, 2, 3, 4, 5], dtype='int64')
>>> s.drop(0, inplace=True)
>>> s.index
Int64Index([1, 2, 3, 4, 5], dtype='int64')
同样适用于数据框(dataframe)。
>>> df = p.DataFrame([np.random.randn(6), np.random.randn(6)])
>>> df.index
Int64Index([0, 1], dtype='int64')
>>> df.drop(0, inplace=True)
>>> df.index
Int64Index([1], dtype='int64')