python - 通过索引中的部分字符串匹配选择行

Question

有这样一个系列:

ds = Series({'wikipedia':10,'wikimedia':22,'wikitravel':33,'google':40})

google        40
wikimedia     22
wikipedia     10
wikitravel    33
dtype: int64

我想选择“wiki”是索引标签(部分字符串标签)一部分的行。

目前我试过了

ds[ds.index.map(lambda x: 'wiki' in x)]

wikimedia     22
wikipedia     10
wikitravel    33
Name: site, dtype: int64

它完成了工作，但不知何故，索引像列所具有的那样要求“包含”...

有什么更好的方法吗？

Answer 1

一个有点厚颜无耻的方式可能是使用loc:

In [11]: ds.loc['wiki': 'wikj']
Out[11]:
wikimedia     22
wikipedia     10
wikitravel    33
dtype: int64

这基本上等同于 ds[ds.index.map(lambda s: s.startswith('wiki'))]。

正如@DSM 所建议的那样，To do contains 可能更好地写成:

ds[['wiki' in s for s in ds.index]]