python - 将随机日期添加到时间戳值存储在列中并将结果保存在新列中

Question

我在一个列中存储了多个入院日期，我想在新列中添加入院日期后 3 到 20 天之间的 future 随机日期(出院日期)。

我的入学日期:

show = pd.DataFrame(columns=[ 'admission dates'])

show['admission dates'] = random_dates(start=pd.to_datetime('2021-01-01'), 
                                             end=pd.to_datetime('2021-01-31'), size=5)

表演

admission dates
0   2021-01-02
1   2021-01-16
2   2021-01-10
3   2021-01-27
4   2021-01-30

我想要得到什么

admission dates Discharge dates
0   2021-01-02  2021-01-05      # +3
1   2021-01-16  2021-01-26      # +10 
2   2021-01-10  2021-01-15      # +5
3   2021-01-27   ...
4   2021-01-30

Answer 1

我们可以使用 np.random.randint 生成一个介于 3 到 20 之间的随机整数，然后将生成的整数的 dtype 更改为 timedelta64 并加上对应的值入学日期

dates = np.random.randint(3, 20, len(show)).astype('timedelta64[D]')
show['discharge date'] = show['admission dates'] + dates

---

>>> dates
array([14,  9,  6, 16,  6], dtype='timedelta64[D]')

>>> show
  admission dates discharge date
0      2021-01-02     2021-01-16 # + 14 days
1      2021-01-16     2021-01-25 # + 9 days
2      2021-01-10     2021-01-16
3      2021-01-27     2021-02-12
4      2021-01-30     2021-02-05 # + 6 days