pandas apply使用多列計算生成新的列實現示例

在python數據分析中，有時需要根據多列數據生成中間結果，pandas給我們帶來瞭很多方便，通常簡短的代碼可以實現一些高級功能，靈活掌握一些技巧可以事倍功半

pandas的apply方法用於對指定列的每個元素進行相同的操作，下面生成一個dataFrame用於演示：

import pandas as pd
a=range(5)
b=range(5,10)
c=range(10,15)
data=pd.DataFrame([a,b,c]).T
data.columns=["a","b","c"]
print(data)

上面的代碼生成的數據如下：

a b c
0 0 5 10
1 1 6 11
2 2 7 12
3 3 8 13
4 4 9 14

下面使用使用a,b兩列相加生成x1列

data["x1"]=data[["a","b"]].apply(lambda x:x["a"]+x["b"],axis=1)

結果如下：

   a b   c x1
0 0 5 10   5
1 1 6 11   7
2 2 7 12   9
3 3 8 13 11
4 4 9 14 13

關鍵的參數是axis=1，指定計算的方向是行而不是列，默認是0，也就是按列進行計算

到此這篇關於pandas apply使用多列計算生成新的列實現示例的文章就介紹到這瞭,更多相關pandas apply多列計算生成新的列內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet！