如何从pandas中的多个列计算多个列

Question

我正在尝试使用一个函数从pandas数据帧中的多个列计算多个列。该函数接受三个参数-a-、-b-和-c-，并返回三个计算值-sum-、-prod-和-quot-。在我的pandas数据框架中，我有三个列-a-、-b-和-c-我想从中计算列-sum-、-prod-和-quot-。

我所做的映射只有在正好有三行时才起作用。我不知道出了什么问题，尽管我认为这与选择正确的轴有关。有人能解释一下发生了什么，以及我如何计算我想要的值吗。下面是我测试过的情况。

初始值

def sum_prod_quot(a,b,c):
    sum  = a + b + c
    prod = a * b * c
    quot = a / b / c
    return (sum, prod, quot)

df = pd.DataFrame({ 'a': [20, 100, 18],
                    'b': [ 5,  10,  3],
                    'c': [ 2,  10,  6],
                    'd': [ 1,   2,  3]
                 })

df
    a   b   c  d
0   20   5   2  1
1  100  10  10  2
2   18   3   6  3

计算步骤

正好使用三行

当我从这个数据框中计算三列并使用函数时，我得到：

df['sum'], df['prod'], df['quot'] = \
        list( map(sum_prod_quot, df['a'], df['b'], df['c']))

df
     a   b   c  d    sum     prod   quot
0   20   5   2  1   27.0    120.0   27.0
1  100  10  10  2  200.0  10000.0  324.0
2   18   3   6  3    2.0      1.0    1.0

这正是我想要的结果：总和列具有a、b、c列中元素的总和；prod-列具有a、b、c列中元素的乘积，而quot-列具有a、b、c列中的元素。

使用三行以上

当我用一行展开数据帧时，我得到一个错误！

数据帧定义为：

df = pd.DataFrame({ 'a': [20, 100, 18, 40],
                    'b': [ 5,  10,  3, 10],
                    'c': [ 2,  10,  6,  4],
                    'd': [ 1,   2,  3,  4]
                 })
df
     a   b   c  d
0   20   5   2  1
1  100  10  10  2
2   18   3   6  3
3   40  10   4  4

电话是

df['sum'], df['prod'], df['quot'] = \
        list( map(sum_prod_quot, df['a'], df['b'], df['c']))

结果是

...
    list( map(sum_prod_quot, df['a'], df['b'], df['c']))
ValueError: too many values to unpack (expected 3) 

虽然我希望多排一行：

df
     a   b   c  d    sum     prod   quot
0   20   5   2  1   27.0    120.0   27.0
1  100  10  10  2  200.0  10000.0  324.0
2   18   3   6  3    2.0      1.0    1.0
3   40  10   4  4   54.0   1600.0    1.0

使用少于三行的

当我用一行减少数据帧时，我也会得到一个错误。数据框定义为：

df = pd.DataFrame({ 'a': [20, 100],
                    'b': [ 5,  10],
                    'c': [ 2,  10],
                    'd': [ 1,   2]
                 })
df
     a   b   c  d
0   20   5   2  1
1  100  10  10  2

电话是

df['sum'], df['prod'], df['quot'] = \
        list( map(sum_prod_quot, df['a'], df['b'], df['c']))

结果是

...
    list( map(sum_prod_quot, df['a'], df['b'], df['c']))
ValueError: need more than 2 values to unpack

虽然我希望少一点争吵：

df
     a   b   c  d    sum     prod   quot
0   20   5   2  1   27.0    120.0   27.0
1  100  10  10  2  200.0  10000.0  324.0

问题

我的问题是：

1） 为什么会出现这些错误？

2） 如何修改调用以获得所需的数据帧？

笔记

在这个链接中，有人问了一个类似的问题，但是给出的答案对我不起作用。

Answer 1

对于3行，答案似乎也不正确。除了第一行和第一列之外，您可以检查其他值吗？查看结果，20*5*2的乘积不是120，而是200，并放置在下面的和列中。在分配给新列之前，您需要以正确的方式形成列表。您可以尝试使用以下设置新列：

df['sum'], df['prod'], df['quot'] = zip(*map(sum_prod_quot, df['a'], df['b'], df['c']))

有关详细信息，请点击链接