如何从组中更改原始DataFrame?

时间:2014-07-07 13:06:52

标签: python pandas pandas-groupby

假设我有以下DataFrame:

import pandas as pd

df = pd.DataFrame({'label': ['a', 'a', 'b', 'b', 'a', 'b', 'c', 'c', 'a', 'a'], 
                   'numbers': [1, 1, 1, 1, 1, 2, 2, 2, 2, 2],
                   'arbitrarydata': [False] * 10})

我想根据两个其他列中的值为arbitrarydata列分配值。一种天真的方法如下:

for _, grp in df.groupby(('label', 'numbers')):
    grp.arbitrarydata = pd.np.random.rand()

当然,这不会将更改传播回df。有没有办法修改组,以便更改反映在原始DataFrame中

1 个答案:

答案 0 :(得分:2)

尝试使用transform,例如:

df['arbitrarydata'] = df.groupby(('label', 'numbers')).transform(lambda x: np.random.rand())