pandas.DataFrame获取满足某条件的行、元素所在的索引(index)

dataframe中根据某些条件得到特定的行或者特定的元素，如何找出这些行、元素所在的index，也就是行号。

_temp = {'job':['farmer', 'teacher', 'worker', 'acter', 'present'], 'money':[3000, 7000, 5000, 100000, 66666]}
df = pd.DataFrame(_temp)
print(df)
>>        job   money
>>0   farmer    3000
>>1  teacher    7000
>>2   worker    5000
>>3    acter  100000
>>4  present   66666
a = df[(df['money']>10000)].index.tolist()
print(a)
>>[3, 4]

流程如下：

确定一个条件–> df[‘money’]>10000)

确定满足条件的行–> df[(df[‘money’]>10000)]

确定这些行的index–> df[(df[‘money’]>10000)].index.tolist()

返回的是一个list

• df.set_index(‘col_name’)设置index

顺便说一下，设置某一列为dataframe的index

_temp = {'job':['farmer', 'teacher', 'worker', 'acter', 'present'], 'money':[3000, 7000, 5000, 100000, 66666]}
df = pd.DataFrame(_temp)
df = df.set_index('job')
print(df)
>>          money
>>job            
>>farmer     3000
>>teacher    7000
>>worker     5000
>>acter    100000
>>present   66666

• 参考

1. AI_盲

2. 小白白白又白cdllp

.
.
.
2019-03-15 18:51:39写于杭州