当前位置 : 主页 > 编程语言 > python >

python – 如何在Pandas系列中找到输入数字中最接近的值?

来源:互联网 收集:自由互联 发布时间:2021-06-25
我见过: how do I find the closest value to a given number in an array? How do I find the closest array element to an arbitrary (non-member) number?. 这些与香草Python有关,而不是熊猫. 如果我有这个系列: ix num 0 11 6
我见过:

> how do I find the closest value to a given number in an array?
> How do I find the closest array element to an arbitrary (non-member) number?.

这些与香草Python有关,而不是熊猫.

如果我有这个系列:

ix   num  
0    1
1    6
2    4
3    5
4    2

我输入3,我怎样才能(有效地)找到?

>如果在系列中找到,则索引为3
>如果在系列中找不到,则低于和高于3的值的索引.

IE浏览器.使用上面的系列{1,6,4,5,2}和输入3,我应该得到带有索引(2,4)的值(4,2).

您可以使用argsort()之类的

比如说,输入= 3

In [198]: input = 3

In [199]: df.ix[(df['num']-input).abs().argsort()[:2]]
Out[199]:
   num
2    4
4    2

df_sort是具有2个最接近值的数据帧.

In [200]: df_sort = df.ix[(df['num']-input).abs().argsort()[:2]]

对于索引,

In [201]: df_sort.index.tolist()
Out[201]: [2, 4]

对于价值观,

In [202]: df_sort['num'].tolist()
Out[202]: [4, 2]

细节,对于上面的解决方案df是

In [197]: df
Out[197]:
   num
0    1
1    6
2    4
3    5
4    2
网友评论