根据多行计算Pandas中的列

时间:2017-07-15 21:21:25

标签: python pandas numpy dataframe

我的数据框包含以下列:

  1. 活动
  2. 发生此事件的交易ID
  3. 以前的交易ID
  4. 我想添加一列来计算此事件是否发生在上一个事务中

    index.ios.js

    “Same_Event_in_prev_trans”列的值应为0 1 0 0 0 0 1

    我不知道如果没有for循环怎么做。

    感谢。

3 个答案:

答案 0 :(得分:2)

您可以使用lambda检查先前trans的事件。

items["Same_Event_in_prev_trans"]=(
    items.apply(lambda x: 1 if x.event==items.set_index('transaction_ID')
                                             .loc[x.previous_trans,'event'] 
                            else 0, axis=1)
    )



items
Out[239]: 
  event  previous_trans  transaction_ID  Same_Event_in_prev_trans
0     A               2               1                         0
1     B               3               2                         1
2     B               5               3                         0
3     A               7               4                         0
4     C               4               5                         0
5     C               1               6                         0
6     C               6               7                         1

答案 1 :(得分:2)

不完全确定逻辑,但检查每个事件的 previous_trans 是否在 transaction_ID 集内,似乎提供了所需的输出:

items["Same_Event_in_prev_trans"] = (items.groupby('event', group_keys=False)
                                     .apply(lambda g: g.previous_trans.isin(g.transaction_ID))
                                     .astype(int))
​
items
#   event   previous_trans  transaction_ID  Same_Event_in_prev_trans
#0      A                2              1                       0
#1      B                3              2                       1
#2      B                5              3                       0
#3      A                7              4                       0
#4      C                4              5                       0
#5      C                1              6                       0
#6      C                6              7                       1

答案 2 :(得分:2)

怎么样

>> items['prev_event'] = pd.merge(items, items[['event', 'transaction_ID']], 
>>                                left_on='previous_trans', 
>>                                right_on='transaction_ID')['event_y']
>> items['same_event'] = (items['event'] == items['prev_event']).astype(int)