python - 在 Pandas 的MultiIndex数据框中修改重复的子索引

嗨，我有一个数据框切片如下：

|            |          | Lemon | Orange |
|------------|----------|-------|--------|
| Date       | Location |       |        |
| 01/01/2016 | Park     | 10    | 20     |
| 01/01/2016 | Beach    | 5     | 15     |
| 01/01/2016 | Park     | 2     | 4      |
| 02/01/2016 | Park     | 8     | 3      |

如您所见，(01/01/2016, Park)有一个重复项，原因是在Park中，第3个条目在k后面有一个空格。我在有限的索引选择技能上遇到了困难，无法对整个“位置”列执行rstrip(" ")以避免空白错误。

最终，我希望做一个groupby函数来可视化Park与其他位置之间的数据。目前，"Park"和"Park "是2个不同的位置。

有什么建议吗？

最佳答案

索引是不可变的，因此，如果要更改index标签，则需要设置一个新的index（感谢IanS）。

您可以在第二级使用str.strip通过get_level_values选择：

new_index = list(zip(df.index.get_level_values('Date'),
                     df.index.get_level_values('Location').str.strip()))

df.index = pd.MultiIndex.from_tuples(new_index, names = df.index.names)

print (df.index)
MultiIndex(levels=[[2016-01-01 00:00:00, 2016-02-01 00:00:00], ['Beach', 'Park']],
           labels=[[0, 0, 0, 1], [1, 0, 1, 1]],
           names=['Date', 'Location'])

如果要使用rstrip，请将str.strip替换为str.rstrip。