pandas-一组列和另一组列相加[关闭]

首先创建带组的扁平数据帧,用于按 numpy.ravel 和 numpy.repeat :

a = df.filter(like='price').values.ravel()
b = df.filter(like='amt').values.ravel()
c = np.repeat(np.arange(len(df)), len(df.filter(like='price').columns))
df = pd.DataFrame({'bin':a, 'amt':b, 'g':c})
print (df)
    bin  amt  g
0   451   10  0
1   454    1  0
2   462    2  0
3   470    5  0
4   448    8  1
5   452    2  1
6   458    6  1
7   464    2  1
8   461    1  2
9   463    3  2
10  468    6  2
11  480    9  2
12  453    4  3
13  455    3  3
14  471    2  3
15  481    4  3

然后通过 cut -我尝试动态创建标签和箱子按楼层划分和多个按 10 ,然后汇总 sum 被重塑 unstack :

val = (df['bin'] // 10)
labels = np.arange(val.min() * 10, val.max() * 10 + 10, 10)
bins = np.append(labels, val.max() * 10 + 10)

df = (df.groupby(['g', pd.cut(df['bin'], bins=bins, labels=labels, right=False)])['amt'].sum()
       .unstack(fill_value=0))
print (df)
bin  440  450  460  470  480
g                           
0      0   11    2    5    0
1      8    8    2    0    0
2      0    0   10    0    9
3      0    7    0    2    4

上次创建执行的格式 df -添加新列依据 assign 和 join 起初的 东风 使用重命名的列:

cols1 = ['bin{}'.format(x) for x in range(1, len(df.columns) + 1)]
cols2 = ['amt{}'.format(x) for x in range(1, len(df.columns) + 1)]

d1= dict(zip(cols1, df.columns))
d2= dict(zip(df.columns, cols2))

df1 = pd.DataFrame(index=df.index).assign(**d1).join(df.rename(columns=d2))
print (df1)
   bin1  bin2  bin3  bin4  bin5  amt1  amt2  amt3  amt4  amt5
g                                                            
0   440   450   460   470   480     0    11     2     5     0
1   440   450   460   470   480     8     8     2     0     0
2   440   450   460   470   480     0     0    10     0     9
3   440   450   460   470   480     0     7     0     2     4