pandas：根据条件获取元素所在的位置（索引）.index.tolist()

在dataframe中根据一定的条件，得到符合要求的某行元素所在的位置。

df = pd.DataFrame({'BoolCol': [1, 2, 3, 3, 4],'attr': [22, 33, 22, 44, 66]},
       index=[10,20,30,40,50])
print(df)
a = df[(df.BoolCol==3)&(df.attr==22)].index.tolist()
print(a)

df如下所示，以上通过选取“BoolCol”取值为3且“attr”取值为22的行，得到该行在df中的位置

    BoolCol  attr
10        1    22
20        2    33
30        3    22
40        3    44
50        4    66
[30]

_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

data={'time':Time,
      'weight':weight}

ss=pd.DataFrame(data)
minn=ss[ss.weight==ss.weight.min()]
print(minn)  
'''
         time  weight
15 2019-09-18    2.3
'''
print(str(minn.values[0]) )  
'''
[Timestamp('2019-09-18 00:00:00') 2.3]
'''