对每个组进行分组并用一定范围内的随机数替换某些值

时间:2019-10-21 21:19:57

标签: python pandas numpy random pandas-groupby

问题是有关根据某些条件为groupby中每个组的唯一值更新列的值。

我有一个这样的数据框:

import numpy as np
import pandas as pd

df = pd.DataFrame({'match_id': ['m1', 'm1', 'm1', 'm1', 'm1', 'm1', 'm2', 'm2', 'm2', 'm2', 'm2', 'm2', 'm3', 'm3', 'm3', 'm3'],
                   'name':['peter', 'mike', 'jeff', 'john', 'alex', 'joe', 'jeff', 'peter', 'alex', 'li', 'joe', 'tom', 'mike', 'john', 'tom', 'peter'],
                   'rank': [3, 3, 3, 3, 1, 2, 1, 2, 4, 2, 3, 2, 1, 2, 3, 2],
                  'rating': [1500, 1500, 1500, 1500, 1550, 1540, 1640, 1500, 1390, 1500, 1450, 1500, 1720, 1500, 1320, 1500]})

我需要根据另一列的条件为“ match_id”中的每组值修改一些数字。

因此,我首先对match_id进行了分组。现在,对于“等级”列中的每1500个值,我想将“等级”列中的相应值更新为1到相应组的长度的范围内的值,该长度在该组中也是不正确的。

这是我到目前为止所做的:


new = pd.DataFrame()
grouped = df.groupby('match_id', sort=False)
for name, dfg in grouped:
    dfm = dfg.copy()
    num = len(dfm.loc[dfm['rating'] == 1500])
    dfm.loc[dfm['rating'] == 1500, 'rank'] = np.random.choice(range(1,len(dfm)+1), num, replace=False)
    new = pd.concat([new, dfm], sort = True)

这有效,但是有两个问题。首先,以这种方式生成的数字可能已经存在于组中(在其他行上)。我希望生成的随机数是唯一的,这意味着相应组中不存在这些数字。

第二,这对于我的原始数据集(125000组)而言花费的时间太长。因此,我需要它比loc更快,更高效。

这是我期望得到的输出(请注意“等级”列)

    match_id    name    rank    rating
0         m1    peter      6    1500
1         m1    mike       5    1500
2         m1    jeff       4    1500
3         m1    john       3    1500
4         m1    alex       1    1550
5         m1    joe        2    1540
6         m2    jeff       1    1640
7         m2    peter      2    1500
8         m2    alex       4    1390
9         m2    li         5    1500
10        m2    joe        3    1450
11        m2    tom        6    1500
12        m3    mike       1    1720
13        m3    john       4    1500
14        m3    tom        3    1320
15        m3    peter      2    1500

我们非常感谢您的帮助。

0 个答案:

没有答案