在熊猫数据框中应用功能

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数据框（然后在执行循环以将其恢复为原始格式后再次转置它）

编辑

实际需要的是两列的前几行中的元素与两列的当前行中的元素的乘积。 这可以使用shift来完成：

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)