优化比较DataFrames的Pandas功能

时间:2018-05-02 14:04:07

标签: python pandas dataframe

我有交易记录日志,用于记录自助服务终端机器的使用情况,以及另一组用于机器在线/离线时间的日志。事务日志包含一个日期时间字段,可以让您知道事务(或会话)何时发生。

    event_date  raw_data1   session_id  ws_id
0   2017-11-06 12:13:06 {'description': 'Home'} 0604e80d-1ae6-48d0-81bf-32ca1dc58e4c    machine2
1   2017-11-06 12:13:41 {'description': 'AreYouStillThere'} 0604e80d-1ae6-48d0-81bf-32ca1dc58e4c    machine2
2   2017-11-06 12:14:09 {'description': 'AttractiveAnimation'}  0604e80d-1ae6-48d0-81bf-32ca1dc58e4c    machine2
3   2017-11-07 10:06:15 {'description': 'Home'} e2e7565f-60b4-4e7b-a8f0-d0a9c384b283    machine13
4   2017-11-07 10:06:27 {'description': 'AuthenticationPanelAdmin'} e2e7565f-60b4-4e7b-a8f0-d0a9c384b283    machine13

此功能的目标是查看哪个session_ids与离线日志一致

    dtrange start   end status  machine_id
0   DateTimeTZRange(datetime.datetime(2017, 11, 17...   2017-11-17 14:46:15 2017-11-17 15:01:15 2   12
1   DateTimeTZRange(datetime.datetime(2017, 11, 17...   2017-11-17 14:47:02 2017-11-17 15:02:02 2   22
2   DateTimeTZRange(datetime.datetime(2017, 11, 17...   2017-11-17 14:47:23 2017-11-17 15:02:23 2   18
3   DateTimeTZRange(datetime.datetime(2017, 11, 17...   2017-11-17 14:48:09 2017-11-17 15:03:09 2   17
4   DateTimeTZRange(datetime.datetime(2017, 11, 17...   2017-11-17 14:49:18 2017-11-17 15:04:18 2   15

ws_id和machine_id是相同的,这使得它变得有点棘手,因为会话时间和machine_id必须在两个数据帧之间匹配。

这是我用来返回机器离线时发生的所有session_ids的代码。它使用事务数据帧中的每一行过滤离线数据帧,如果离线事件与会话时间一致,则返回session_id:

def CheckSession(machinename, sessiontime, sessionid):
    if len(offlinedf[(offlinedf.start<sessiontime)
             &(offlinedf.end>sessiontime)
             &(offlinedf.name==machinename)])>0:
        return sessionid

sessions = df.apply(lambda row: CheckSession(row["name"], row["created_at1"], row["session_id"]), axis=1)

这会构建会话列表,但速度非常慢且数据帧非常大。我还在学习如何最好地使用pandas库 - 我希望使用一些矢量化来优化它,但是却无法弄清楚如何以这种方式构建它。

1 个答案:

答案 0 :(得分:1)

通过 name 考虑merging df offlinedf ,然后根据您内部的逻辑使用query进行过滤功能。然后将过滤后的数据框的 sessionid 列转换为列表。

session_df = df.merge(offlinedf, on='name', suffixes=['', '_'])\
               .query('start < created_at1 & end > created_at1') 

sessions = session_df['sessionid'].tolist()

在任何数据分析工作中,对象的逐块处理优于迭代行处理。