从某个范围内的列中找到一个固定值，并在熊猫数据框中找到另一列的每个唯一值

Question

我有一个像这样的数据框：

df
col1      col2
 1        50000
 1        2000
 2        51000
 3        100
 3        5000
 3        50500
 4        200
 4        51500
 5        49000

我想为每个col1唯一值出现的每个col1值标识正负10％的值。

最终输出应如下所示

col1        col2
  1         50000
  2         51000
  3         50500
  4         51500
  5         49000

如果存在除50000左右以外的其他值并且在正负10％范围内，则将那些具有50000左右的值相加

如何以最有效的方式使用pandas / python？

Answer 1

使用列表cpmprehension for循环的所有唯一值col2通过，过滤器+-10%与Series.between和boolean indexing ，如果各组存在通过创建集中的所有值进行比较col1 。 按Series.isin最后一个过滤器：

s = set(df['col1'])
print (s)
{1, 2, 3, 4, 5}

a = [x for x in df['col2'].unique() 
     if set(df.loc[df['col2'].between(x - x *.1, x + x*.1), 'col1']) == s]
print (a)
[50000, 51000, 50500, 51500, 49000]

df = df[df['col2'].isin(a)]
print (df)
   col1   col2
0     1  50000
2     2  51000
5     3  50500
7     4  51500
8     5  49000