在Python中查找满足条件的列中的第一行(值大于x)

时间:2014-03-22 12:41:49

标签: python arrays numpy pandas

在Python中,(+ Numpy或Pandas?)

我正在寻找一种方法来查找值高于X的第一行(假设 9 )。

1 个答案:

答案 0 :(得分:0)

不确定[2,10,5]行的来源。但总体思路就像......

data = {'values':[[1,10,1],[3,1,3],[4,3,5],[6,6,7],[7,10,8],[7,10,16],[14,5,22],[18,1,30]]}
x = 9
for i,row in enumerate(data['values']):
    if row[2]>x:
        rows = data['values'][max(0,i-4):i]
        break

assert rows == [[3,1,3],[4,3,5],[6,6,7],[7,10,8]]

# etc...
y = 8
for i,row in enumerate(rows):
    if row[1]>y:
        ans = rows[i][0]
        break

# etc...
assert ans == 7