我有一个像这样创建的系列
s = pd.Series({'a': False, 'b': True, 'c': False, 'd': True, 'e': False})
>> s
a False
b True
c False
d True
e False
dtype: bool
有没有办法整齐地提取它的真实名称,保留在Pandas或NumPy中而不回到普通的Python?
目前我正在使用它:
sdict = s.to_dict()
for item in list(sdict):
if sdict[item] == True:
print (item, end=" ")
>> b d
答案 0 :(得分:2)
将boolean indexing
与s.index
:
print (s.index[s])
Index(['b', 'd'], dtype='object')
print (s.index[s].tolist())
['b', 'd']
print (', '.join(s.index[s]))
b, d
带有np.where
的有点过于复杂的解决方案,为了好玩:
print (s.index[np.where(s)[0]])
Index(['b', 'd'], dtype='object')