如何从pd.Dataframe中提取索引

Question

我有一些Dataframe

import numpy as np
import pandas as pd
df = pd.DataFrame({
    'name':  ['Alice','John','Peter','Richard'],
    'age':  [23,28,43,29],
    'gender': [0,1,1,1],
    'salary': [900, 400, 900, 400]
})

我可以从这个

中提取任何值

df.ix[df['name'] == 'Alice', 'salary']
0    900
Name: salary, dtype: int64

虽然我想只提取索引

0
df.ix[df['name'] == 'Alice', index] 
df.ix[df['name'] == 'Alice', df.index]

我收到错误

如何解决？

Answer 1

使用：

User-Agent

如果需要输出print (df[df['name'] == 'Alice'].index) Int64Index([0], dtype='int64') print (df.index[df['name'] == 'Alice']) Int64Index([0], dtype='int64') ：

list

如果需要选择list的第一个值：

print (df[df['name'] == 'Alice'].index.tolist())
[0]