python - 减少数据框行和查找

Question

需要以下方面的帮助。假设我们有一个数据框:

dictionary ={'Category':['a','a','a','a','a','a','a','a','b','b','b','b','b','b','b'], 
'val1':[11,13,14,17,18,21,22,25,2,8,9,13,15,16,19],
'val2':[1,0,5,1,4,3,5,9,4,1,5,2,4,0,3]}

df=pd.DataFrame(dictionary)

“val1”始终在“category”中的相同值内递增，即类别的第一行和最后一行是该类别的最小值和最大值。每个类别的行太多，我想创建一个新的数据框，其中包含每个类别的 min 和 max 值，并包含等间距的例如每个类别 5 行(包括最小值和最大值)。我认为 numpy 的 linspace 应该用于为每个类别创建一个值数组(例如 linspace(min, max, 5))，然后类似于 excel 的“查找”功能应该用于从 df 获取最接近的“val1”值。或者也许还有其他更好的方法......

非常感谢您的帮助。

Answer 1

这是您需要的吗？使用groupby和reindex

l=[]
for _, x in df.groupby('Category'):
    x.index=x['val1']
    y=x.reindex(np.linspace(x['val1'].min(), x['val1'].max(), 5),method='nearest')
    l.append(y)
pd.concat(l)
Out[330]: 
      Category  val1  val2
val1                      
11.00        a    11     1
14.50        a    14     5
18.00        a    18     4
21.50        a    22     5
25.00        a    25     9
2.00         b     2     4
6.25         b     8     1
10.50        b     9     5
14.75        b    15     4
19.00        b    19     3