pandas使用另一个数据框列

时间:2018-05-01 19:49:58

标签: python pandas pandas-groupby

有两个数据框,两个数据框都有共同的user_id。让我们称他们为df_Adf_Bdf_B每个用户有一条记录。 df_A包含每个用户的多条记录,但对于每个用户,都存在一条NaN的记录。

DF_A

    user_id     start_time       time_calc
0     421    2015-10-25 11:42:35      8.97
1     421    2015-10-25 11:51:33      4.85
2     421    2015-10-25 11:56:24      6.77
3     421    2015-10-25 13:03:10       NaN
4     493    2015-10-26 05:08:13      0.85
5     493    2015-10-26 05:09:04       NaN
6     636    2015-10-17 15:30:53      2.55
7     636    2015-10-17 15:33:26      4.17
8     636    2015-10-17 15:37:36      2.35
9     636    2015-10-17 15:39:57      0.32
10    636    2015-10-17 15:40:16      0.43
11    636    2015-10-17 15:40:42      0.65
12    636    2015-10-17 15:41:21       NaN

DF_B

   user_id        last_time
0    100     2015-11-04 01:52:16
1    421     2015-10-25 14:03:23
2    493     2015-10-26 05:09:51
3    636     2015-11-03 20:15:54
4    030     2015-11-03 17:44:20
5    174     2015-10-10 00:20:37
6    437     2015-10-24 12:34:38
7    481     2015-10-28 03:18:24
8    952     2015-10-28 21:07:15
9    197     2015-10-18 14:52:34

目标是通过执行NaN填充df_A中的df_B.last_time - df_A.start_time,但仅限Nan,同时使用正确的user_id来匹配记录。< / p> 逻辑

:使用以下数据填写user_id 421 NaN记录: 2015-10-25 14:03:23 - 2015-10-25 13:03:10

df_A.groupby('user_id').time_calc.fillna(df_B.last_time - df_A.time_calc.isnull['start_time'])

不幸的是,这没有用。

1 个答案:

答案 0 :(得分:2)

首先,使用merge作为第一步构建填充值映射器:

i = df_A[df_A.time_calc.isnull()].merge(df_B)
j = i['last_time'].sub(i['start_time']).dt.total_seconds() / 60
j.index = i.user_id

j
user_id
421       60.216667
493        0.783333
636    24754.550000
dtype: float64

现在,做一个fillna

df_A['time_calc'] = df_A['time_calc'].fillna(df_A.user_id.map(j))  
df_A

    user_id          start_time     time_calc
0       421 2015-10-25 11:42:35      8.970000
1       421 2015-10-25 11:51:33      4.850000
2       421 2015-10-25 11:56:24      6.770000
3       421 2015-10-25 13:03:10     60.216667
4       493 2015-10-26 05:08:13      0.850000
5       493 2015-10-26 05:09:04      0.783333
6       636 2015-10-17 15:30:53      2.550000
7       636 2015-10-17 15:33:26      4.170000
8       636 2015-10-17 15:37:36      2.350000
9       636 2015-10-17 15:39:57      0.320000
10      636 2015-10-17 15:40:16      0.430000
11      636 2015-10-17 15:40:42      0.650000
12      636 2015-10-17 15:41:21  24754.550000
相关问题