Расчет дневной суммы в панде - PullRequest
Новая
       0

Расчет дневной суммы в панде

2 голосов
/ 19 сентября 2019

У меня есть кадр данных, который выглядит ниже, и я пытаюсь рассчитать суточную дозу для каждого лекарства.В приведенной ниже таблице у меня есть один препарат, но в фактической таблице у меня есть несколько препаратов.

Позвольте мне дать краткое объяснение таблицы и сценария.Столбец идентификатора - это уникальный идентификатор пациента. В столбце «Начало» и «Конец» указано, когда лекарство началось и закончилось для данной частоты, в столбце «Часы» указано количество часов между началом и концом для данного идентификатора. В столбце «Частота» указано, с какой скоростьюлекарство давалось в течение этого периода времени для этого удостоверения личности.Таблица составлена ​​таким образом, что для каждого показателя для каждого показателя предусмотрена дата начала и окончания.

В больничных условиях, если они хотят рассчитать общую суточную дозу, они рассчитывают с 7 часов утра сегодня до 7 часов.Я на следующий день вместо того, чтобы делать это с полуночи до полуночи.

  ID           START                   END            DRUG     RATE  HOURS
15114299  2011-03-18 18:23:00  2011-03-19 20:59:59  morphine   0.03   26.6
15114299  2011-03-19 21:00:00  2011-03-20 01:29:59  morphine    0.1   4.4
15340823  2011-02-18 16:00:00  2011-02-19 03:09:59  morphine    0.1   11.16
15340823  2011-02-19 15:30:00  2011-02-19 21:59:59  morphine    0.05  6.49
15621352  2011-04-20 03:44:00  2011-04-20 13:41:59  morphine    0.05  9.96
15621352  2011-04-20 13:42:00  2011-04-20 23:59:59  morphine    0.5   10.29
15775791  2011-07-21 19:00:00  2011-07-24 03:59:59  morphine    0.1   56.99
15775791  2011-07-24 04:00:00  2011-07-24 14:14:59  morphine    0.05  10.24
15947559  2011-07-05 19:40:00  2011-07-06 05:43:59  morphine    0.1   10.06
15947559  2011-07-06 05:44:00  2011-07-09 01:59:59  morphine    0.15  68.26
15947559  2011-07-09 02:00:00  2011-07-09 18:59:59  morphine    0.1   16.99
15947559  2011-07-14 19:30:00  2011-07-15 18:29:59  morphine    0.1   22.99
15947559  2011-07-15 18:30:00  2011-07-17 02:59:59  morphine    0.15  32.49
15947559  2011-07-17 03:00:00  2011-07-17 08:59:59  morphine    0.1   5.99
15947559  2011-07-17 09:00:00  2011-07-17 16:59:59  morphine    0.075 7.99

Как мы можем видеть на примере кадра данных, первое лекарство было в течение 26,6 часов, и я хочу разделить эту запись на две части, как показано ниже, и иметь столбец дня, который будет полезен длясгруппировать и получить сумму за каждый день в конце.

Ожидаемый DF: 

  ID           START                   END            DRUG     RATE     HOURS   Days
15114299  2011-03-18 18:23:00  2011-03-19 06:59:59  morphine   0.03     12.62   Day1
15114299  2011-03-19 07:00:00  2011-03-19 20:59:59  morphine   0.03     14      Day2
15114299  2011-03-19 21:00:00  2011-03-20 01:29:59  morphine    0.1     4.5     Day2

15340823  2011-02-18 16:00:00  2011-02-19 03:09:59  morphine    0.1     11.16   Day1
15340823  2011-02-19 15:30:00  2011-02-19 21:59:59  morphine    0.05    6.49    Day2

15621352  2011-04-20 03:44:00  2011-04-20 06:59:59  morphine    0.05    3.27    Day1
15621352  2011-04-20 07:00:00  2011-04-20 13:41:59  morphine    0.05    6.70    Day2
15621352  2011-04-20 13:42:00  2011-04-20 23:59:59  morphine    0.5     10.29   Day2

15775791  2011-07-21 19:00:00  2011-07-22 06:59:59  morphine    0.1     12      Day1
15775791  2011-07-22 07:00:00  2011-07-23 06:59:59  morphine    0.1     24      Day2        
15775791  2011-07-23 07:00:00  2011-07-24 03:59:59  morphine    0.1     21      Day3
15775791  2011-07-24 04:00:00  2011-07-24 06:59:59  morphine    0.05    3       Day3
15775791  2011-07-24 07:00:00  2011-07-24 14:14:59  morphine    0.05    7.25    Day4

15947559  2011-07-05 19:40:00  2011-07-06 05:43:59  morphine    0.1     10.06   Day1
15947559  2011-07-06 05:44:00  2011-07-06 06:59:59  morphine    0.15    1.27    Day1 
15947559  2011-07-06 07:00:00  2011-07-07 06:59:59  morphine    0.15    24      Day2
15947559  2011-07-07 07:00:00  2011-07-08 06:59:59  morphine    0.15    24      Day3
15947559  2011-07-08 07:00:00  2011-07-09 01:59:59  morphine    0.15    19      Day4
15947559  2011-07-09 02:00:00  2011-07-09 06:59:59  morphine    0.1     5       Day4
15947559  2011-07-09 07:00:00  2011-07-09 18:59:59  morphine    0.1     12      Day5
15947559  2011-07-14 19:30:00  2011-07-15 06:59:59  morphine    0.1     11.50   Day6
15947559  2011-07-15 07:00:00  2011-07-15 18:29:59  morphine    0.1     11.50   Day7
15947559  2011-07-15 18:30:00  2011-07-16 06:59:59  morphine    0.15    12.50   Day7
15947559  2011-07-16 07:00:00  2011-07-17 02:59:59  morphine    0.15    20      Day8
15947559  2011-07-17 03:00:00  2011-07-17 06:59:59  morphine    0.1     4       Day8
15947559  2011-07-17 07:00:00  2011-07-17 08:59:59  morphine    0.1     2       Day9
15947559  2011-07-17 09:00:00  2011-07-17 16:59:59  morphine    0.075   8       Day9

(я создал пробелы между двумя идентификаторами просто для большей ясности и понимания)

Я попробовал следующий код.

test = pd.concat([pd.DataFrame({'START': pd.date_range(row['START'], row['END'], freq='D'),
               'PERSON_ID': row['ID'],
               'DRUG': row['DRUG'], 
               'RATE': row['RATE']}, columns=['ID', 'START', 'DRUG', 'RATE'])
           for i, row in df.iterrows()], ignore_index=False)

но это создает фрейм данных следующим образом.

ID              START          DRUG       RATE
15114299    2011-03-18 18:23:00 morphine    0.03
15114299    2011-03-19 18:23:00 morphine    0.03
15114299    2011-03-19 21:00:00 morphine    0.1
15340823    2011-02-18 16:00:00 morphine    0.1
15340823    2011-02-19 15:30:00 morphine    0.05
15621352    2011-04-20 03:44:00 morphine    0.05
15621352    2011-04-20 13:42:00 morphine    0.5
15775791    2011-07-21 19:00:00 morphine    0.1
15775791    2011-07-22 19:00:00 morphine    0.1
15775791    2011-07-23 19:00:00 morphine    0.1
15775791    2011-07-24 04:00:00 morphine    0.05
15947559    2011-07-05 19:40:00 morphine    0.1
15947559    2011-07-06 05:44:00 morphine    0.15
15947559    2011-07-07 05:44:00 morphine    0.15
15947559    2011-07-08 05:44:00 morphine    0.15
15947559    2011-07-09 02:00:00 morphine    0.1
15947559    2011-07-14 19:30:00 morphine    0.1
15947559    2011-07-15 18:30:00 morphine    0.15
15947559    2011-07-16 18:30:00 morphine    0.15
15947559    2011-07-17 03:00:00 morphine    0.1
15947559    2011-07-17 09:00:00 morphine    0.075

Буду очень признателен, если кто-нибудь сможет мне помочь в решении этой проблемы.Рады предоставить дополнительную информацию.

1 Ответ

2 голосов
/ 20 сентября 2019

Начните с определения таких переменных: 

h7 = pd.Timedelta('7H')
s1 = pd.Timedelta('1S')

(будет использоваться несколько раз).

Чтобы «развернуть» каждую строку в последовательность строк (для каждого дня),определите следующую функцию: 

def expand(row):
    tm1, tm2 = row.START, row.END
    t1 = tm1 if type(tm1).__name__ == 'Timestamp' else pd.to_datetime(tm1)
    t2 = tm2 if type(tm2).__name__ == 'Timestamp' else pd.to_datetime(tm2)
    rng = pd.date_range(start=t1 - h7, end=t2 - h7, freq='1D', closed='right',
        normalize=True) + pd.Timedelta('7H')
    ind = pd.DatetimeIndex([t1, *rng, t2 + s1])
    res = pd.DataFrame({'ID': row.ID, 'START': ind[:-1], 'END': ind[1:] - s1,
        'DRUG': row.DRUG, 'RATE': row.RATE})
    res['HOURS'] = (res.END - res.START).dt.total_seconds().div(3600).round(2)
    return res

и примените ее, создав таблицу результатов: 

df2 = pd.concat(df.apply(expand, axis=1).tolist(), ignore_index=True)

на данный момент, без столбца Days .

Затем добавьте столбец Дни , выполнив: 

df2['Days'] = 'Day' + ((df2.START - h7).dt.date - df2.groupby('ID')
    .START.transform('first').dt.date).dt.days.add(1).astype(str)

Номер дня (после «День» ) - это разница между текущей датой и датой изпервая строка + 1.

Из-за вычитания h7 из текущей даты этот код работает корректно, даже если изменение произошло между серединой ночи и 7 часами.

Чтобы представить результат, выполните: 

df2.sort_values('ID')

Чтобы проследить идею этого решения, запустите каждый оператор отдельно и посмотрите результат.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.

Нет похожих вопросов

...