Python Pandas使用相对订单号添加列

时间:2015-03-03 14:22:38

标签: python pandas data-analysis

如何将订单号列添加到现有数据框?

这是我的DataFrame:

import pandas as pd
import math
frame = pd.DataFrame([[1, 4, 2], [8, 9, 2], [10, 2, 1]], columns=['a', 'b', 'c'])

def add_stats(row):
    row['sum'] = sum([row['a'], row['b'], row['c']])
    row['sum_sq'] = sum(math.pow(v, 2) for v in [row['a'], row['b'], row['c']])
    row['max'] = max(row['a'], row['b'], row['c'])
    return row

frame = frame.apply(add_stats, axis=1)
print(frame.head())

结果数据是:

    a  b  c  sum  sum_sq  max
0   1  4  2    7      21    4
1   8  9  2   19     149    9
2  10  2  1   13     105   10

首先,我想添加3个带有订单号的额外列,分别对sum,sum_sq和max进行排序。接下来,这3列应该组合成一列 - 订单号的平均值 - 但我知道如何做这个部分(使用apply和axis = 1)。

1 个答案:

答案 0 :(得分:4)

我认为你正在寻找rank,你提到sorting。举个例子,添加:

frame['sum_order'] = frame['sum'].rank()
frame['sum_sq_order'] = frame['sum_sq'].rank()
frame['max_order'] = frame['max'].rank()

frame['mean_order'] = frame[['sum_order', 'sum_sq_order', 'max_order']].mean(axis=1)

获得:

    a  b  c  sum  sum_sq  max  sum_order  sum_sq_order  max_order  mean_order
0   1  4  2    7      21    4          1             1          1    1.000000
1   8  9  2   19     149    9          3             3          2    2.666667
2  10  2  1   13     105   10          2             2          3    2.333333

秩方法也有一些选项,例如在相同或NA值的情况下指定行为。