在熊貓數據框中應用函數

Question

我已經想出瞭如何將函數應用於熊貓數據框的整個列或子部分，以代替編寫逐個修改每個單元的循環。

在執行上述操作時，是否可以編寫一個將數據框內的單元格作爲輸入的函數？

例如，當前單元格中的函數返回前一個單元格的值乘以該單元格之前的單元格的乘積。我現在正在逐行循環執行此操作，而且這並不令人驚訝，效率很低。我對Python非常陌生。

Answer 1

對於你提到的情況下（前面兩個細胞倍增），你可以做以下的（通過各列循環，但不是每個單元格）：

import pandas as pd 

a = pd.DataFrame({0:[1,2,3,4,5],1:[2,3,4,5,6],2:0,3:0}) 

for i in range(2,len(a)): 
    a[i] = a[i-1]*a[i-2] 

這將使每一列a的前兩列相乘

如果要執行此操作下去行而不是列，你可以transpose數據幀（然後再次執行循環後轉就讓它回到原來的格式）

編輯

實際需要的是前兩行兩列中的元素和兩列當前行中的元素的乘積。這可以通過使用移位：

import pandas as pd 

df= pd.DataFrame({"A": [1,2,3,4], "B": [1,2,3,4], "C": [2,3,4,5], "D": [5,5,5,5]}) 

df['E'] = df['A'].shift(1)*df['B'].shift(1)*df['C']*df['D'] 

df['E'] 

產地：

0  NaN 
1  15.0 
2  80.0 
3 225.0 

Answer 2

這是卓有成效的，並且換擋可以去向前或向後取決於你的需要：

df['Column'] = df['Column'].shift(1) * df['Column'].shift(2)