Skip to main content
 首页 » 编程设计

python之按 X 列对数据框进行分组

2025年05月04日33qq78292959

我有一个数据框,我想对每 2 列(或 3 列,它是可变的)应用一个函数。

例如,对于以下 DataFrame,我想将 mean 函数应用于列 0-1、2-3、4-5、....28-29

d = pd.DataFrame((np.random.randn(360)).reshape(12,30)) 
 
           0         1  ...       17        18        19            29  
0   0.590293 -2.794911 ...  0.772830 -1.389820 -1.696832 ...  0.615549  
1   0.115954  2.179996 ... -0.764384 -0.610713 -0.289050 ... -1.130803  
2   0.209405  0.381398 ... -0.317797  0.261590  2.502581 ...  1.750126  
3   2.828746  0.831299 ... -0.679128 -1.255643  0.245522 ... -0.612011  
4   0.625284  1.141448 ...  0.391047 -1.262303 -0.094523 ... -3.643543  
5   0.493923  1.601924 ... -0.935102 -2.416869  0.112278 ... -0.001863  
6  -1.213347  0.396682 ...  0.671210  0.122041 -1.469256 ...  1.825214  
7   0.026695 -0.482887 ...  0.020123  1.151533 -0.440114 ... -1.407276  
8   0.235436  0.763454 ... -0.446333 -0.322420  1.067925 ... -0.622363  
9   0.668812  0.537556 ...  0.471777 -0.119756  0.098581 ...  0.007390  
10 -1.112536 -2.378293 ...  1.047705 -0.812025  0.771080 ... -0.403167  
11 -0.709457 -1.598942 ... -0.568418 -2.095332 -1.970319 ...  1.687536  

请您参考如下方法:

groupby 也可以在 axis=1 上工作,并且可以接受一系列组标签。如果您的列是方便的范围,就像您的示例中那样,那是微不足道的:

>>> df = pd.DataFrame((np.random.randn(6*6)).reshape(6,6)) 
>>> df 
          0         1         2         3         4         5 
0  1.705550 -0.757193 -0.636333  2.097570 -1.064751  0.450812 
1  0.575623 -0.385987  0.105516  0.820795 -0.464069  0.728609 
2  0.776840 -0.173348  0.878534  0.995937  0.094515  0.098853 
3  0.326854  1.297625  2.232534  1.004719 -0.440271  1.548430 
4  0.483211 -1.182175 -0.012520 -1.766317 -0.895284 -0.695300 
5  0.523011 -1.653557  1.022042  1.201774 -1.118465  1.400537 
>>> df.groupby(df.columns//2, axis=1).mean() 
          0         1         2 
0  0.474179  0.730618 -0.306970 
1  0.094818  0.463155  0.132270 
2  0.301746  0.937235  0.096684 
3  0.812239  1.618627  0.554080 
4 -0.349482 -0.889419 -0.795292 
5 -0.565273  1.111908  0.141036 

(这是有效的,因为 df.columns//2 给出了 Int64Index([0, 0, 1, 1, 2, 2], dtype='int64') .)

即使我们没有那么幸运,我们仍然可以自己建立合适的组:

>>> df.groupby(np.arange(df.columns.size)//2, axis=1).mean() 
          0         1         2 
0  0.474179  0.730618 -0.306970 
1  0.094818  0.463155  0.132270 
2  0.301746  0.937235  0.096684 
3  0.812239  1.618627  0.554080 
4 -0.349482 -0.889419 -0.795292 
5 -0.565273  1.111908  0.141036