Pandas求列的指数和

Pandas求列的指数和

@author: Heisenberg

@date: 2022/11/08

df = pd.DataFrame(np.random.randn(-1,1,size=(5, 4)), columns=list('ABCD'))
# df
#   A  B  C  D
#0 -5  4  2 -4
#1 -2 -1 -1  2
#2  0  4 -3  4
#3  3  2  2  4
#4  1 -3  4  0

#所有列
df['exp_sum']=df.applymap(lambda x: np.exp(x)).sum(axis=1) #指数和
#In [15]: df
#Out[15]:
#   A  B  C  D     exp_sum
#0 -5  4  2 -4   62.012260
#1 -2 -1 -1  2    8.260150
#2  0  4 -3  4  110.246087
#3  3  2  2  4   89.461799
#4  1 -3  4  0   58.366219

#连续几列
df['exp_sum']=df.iloc[:,0:3].applymap(lambda x: np.exp(x)).sum(axis=1)
#   A  B  C  D    exp_sum
#0 -5  4  2 -4  61.993944
#1 -2 -1 -1  2   0.871094
#2  0  4 -3  4  55.647937
#3  3  2  2  4  34.863649
#4  1 -3  4  0  57.366219