【Python】在pandas 的 dateframe 中计算时间差

糊酥饼

已于 2022-11-20 13:21:13 修改

阅读量1.6k

点赞数 2

CC 4.0 BY-SA版权

文章标签： python pandas

于 2022-06-29 23:22:40 首次发布

本文链接：https://blog.csdn.net/weixin_44218218/article/details/125531550

这段代码定义了计算时间区间内早晨和下午时段的函数。它首先将数据框中的日期时间转换为datetime类型，然后根据给定的时间段计算每个记录的时段计数，返回包含不同时段计数的字典。最后，将这些计数添加到原始数据框中作为新列。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

from datetime import datetime
import pandas as pd

df_info = pd.DataFrame([['2022-06-25 9:10:11','2022-06-25 10:14:15'],
['2022-06-25 11:10:11','2022-06-25 15:14:15'],['2022-06-25 12:10:11','2022-06-25 16:25:15']],columns=['join_time','leaving_time'])
df_info['join_time'] = pd.to_datetime(df_info['join_time'],infer_datetime_format=True)
df_info['leaving_time'] = pd.to_datetime(df_info['leaving_time'],infer_datetime_format=True)

# timeStra,timeStrb 皆为字符串对象时间;n 为转换为分、小时、天需要保留几位小数
def getTimeDiff(timeStra, timeStrb, n):
    t1 = str(timeStra)
    t1 = datetime.strptime(t1, "%Y-%m-%d %H:%M:%S")
    t2 = str(timeStrb)
    t2 = datetime.strptime(t2, "%Y-%m-%d %H:%M:%S")
    day_cnt = (t2-t1).days
    second_cnt = (t2-t1).seconds
    all_seconds = day_cnt*24*60*60 + second_cnt
    all_minutes = round(all_seconds/60, n)
    all_hours = round(all_minutes/60, n)
    all_days = round(all_hours/24, n)
    return all_minutes

def cnt_all_time(row):
    time_1 ='2022-06-25 09:10:00'
    time_2 = '2022-06-25 12:00:00'
    time_3 = '2022-06-25 14:00:00'
    cnt = {}
    join_time = str(getattr(row,'join_time'))
    leaving_time = str(getattr(row,'leaving_time'))
    if (join_time > time_1) & (leaving_time <= time_2):
        cnt['morning_time_cnt'] = getTimeDiff(join_time, leaving_time, 2)
        cnt['afternoon_time_cnt'] = 0
        cnt['all_time_cnt'] = cnt['morning_time_cnt']
    if (join_time > time_2) & (leaving_time > time_3):
        cnt['morning_time_cnt'] = 0
        cnt['afternoon_time_cnt'] = getTimeDiff(join_time, leaving_time, 2)
        cnt['all_time_cnt'] = cnt['afternoon_time_cnt'] 
    if (join_time < time_2) & (leaving_time > time_3):
        cnt['morning_time_cnt'] = getTimeDiff(join_time, '2022-06-25 12:00:00', 2)
        cnt['afternoon_time_cnt'] = getTimeDiff(time_3, leaving_time, 2)
        cnt['all_time_cnt'] = cnt['morning_time_cnt'] + cnt['afternoon_time_cnt']
    return  cnt

x=[]
for row in df_info.itertuples():
    row1 = cnt_all_time(row)
    x.append(row1)

df = pd.DataFrame(x)
df_info = pd.concat([df_info, df], axis = 1)
df_info