如何使用np.where进行日期偏移

import numpy as np

dates = np.array(['2022-01-01', '2022-01-02', '2022-01-03'], dtype='datetime64')

offset = np.timedelta64(1, 'D')  # 表示向后偏移1天

new_dates = np.where(condition, dates + offset, dates)

import numpy as np

dates = np.array(['2022-01-01', '2022-01-02', '2022-01-03'], dtype='datetime64')
offset = np.timedelta64(1, 'D')

condition = np.array([True, False, True])  # 假设只对第1和第3个日期进行偏移

new_dates = np.where(condition, dates + offset, dates)
print(new_dates)

['2022-01-02' '2022-01-02' '2022-01-04']