如何在pandas列中找到行索引?

时间:2018-03-18 07:41:18

标签: python pandas

我对pandas非常陌生,并试图让任何index的行value高于lprice。有人能让我快速了解我做错了吗?

数据帧

      StrikePrice
0      40.00
1      50.00
2      60.00
3      70.00
4      80.00
5      90.00
6     100.00
7     110.00
8     120.00
9     130.00
10    140.00
11    150.00
12    160.00
13    170.00
14    180.00
15    190.00
16    200.00
17    210.00
18    220.00
19    230.00
20    240.00

现在我想弄清楚如何获得row index value higher lprice而不是 lprice = 99 for strike in df['StrikePrice']: strike = float(strike) # print(strike) if strike >= lprice: print('The high strike is:' + str(strike)) ce_1 = strike print(df.index['StrikePrice' == ce_1])

0

以上内容将index作为{{1}}

我不确定我在这里做错了什么。

3 个答案:

答案 0 :(得分:4)

在布尔切片后使用index属性。

lprice = 99
df[df.StrikePrice >= lprice].index

Int64Index([6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20], dtype='int64')

如果你坚持迭代并找到你找到它的时间,你可以修改你的代码:

lprice = 99
for idx, strike in df['StrikePrice'].iteritems():
    strike = float(strike)
    # print(strike)
    if strike >= lprice:
        print('The high strike is:' + str(strike))
        ce_1 = strike
        print(idx)

答案 1 :(得分:3)

我认为最好是boolean indexing的过滤器索引:

a = df.index[df['StrikePrice'] >= 99]

#alternative
#a = df.index[df['StrikePrice'].ge(99)]

您的代码应该更改类似:

lprice = 99
for strike in df['StrikePrice']:
    if strike >= lprice:
        print('The high strike is:' + str(strike))
        print(df.index[df['StrikePrice'] == strike])

答案 2 :(得分:2)

如果我们仅指定condition,则

numpy.where(condition[, x, y])会这样做。

np.where()返回元组 condition.nonzero()condition为真的索引,如果只给出condition

In [36]: np.where(df.StrikePrice >= lprice)[0]
Out[36]: array([ 6,  7,  8,  9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20], dtype=int64)

PS感谢@jezrael for the hint - np.where()返回数字索引位置而不是DF索引值:

In [41]: df = pd.DataFrame({'val':np.random.rand(10)}, index=pd.date_range('2018-01-01', freq='9999S', periods=10))

In [42]: df
Out[42]:
                          val
2018-01-01 00:00:00  0.459097
2018-01-01 02:46:39  0.148380
2018-01-01 05:33:18  0.945564
2018-01-01 08:19:57  0.105181
2018-01-01 11:06:36  0.570019
2018-01-01 13:53:15  0.203373
2018-01-01 16:39:54  0.021001
2018-01-01 19:26:33  0.717460
2018-01-01 22:13:12  0.370547
2018-01-02 00:59:51  0.462997

In [43]: np.where(df['val']>0.5)[0]
Out[43]: array([2, 4, 7], dtype=int64)

解决方法:

In [44]: df.index[np.where(df['val']>0.5)[0]]
Out[44]: DatetimeIndex(['2018-01-01 05:33:18', '2018-01-01 11:06:36', '2018-01-01 19:26:33'], dtype='datetime64[ns]', freq=None)