REF: De-duplicate pieces of datetimelike arithmetic #23166
Conversation
|
Hello @jbrockmendel! Thanks for submitting the PR.
|
| result[self._isnan] = fill_value | ||
| return result | ||
|
|
||
| def _nat_new(self, box=True): |
There was a problem hiding this comment.
Only used once outside of tests; less verbose to do inline.
| raise TypeError("{cls} subtraction must have the same " | ||
| "timezones or no timezones" | ||
| .format(cls=type(self).__name__)) | ||
|
|
There was a problem hiding this comment.
The casting and timezone checking used to be done inside the calling function. Cleaner to do it here.
| delta = TimedeltaArrayMixin(delta) | ||
| new_values = self._add_delta_tdi(delta) | ||
| else: | ||
| new_values = self.astype('O') + delta |
| new_values = self._add_delta_td(delta) | ||
| elif is_timedelta64_dtype(delta): | ||
| if not isinstance(delta, TimedeltaArrayMixin): | ||
| delta = TimedeltaArrayMixin(delta) |
There was a problem hiding this comment.
do this inside add_delta_dti
pandas/core/arrays/datetimes.py
Outdated
| if self.tz is not None and self.tz is not utc: | ||
| result = result.tz_convert(self.tz) | ||
| return result | ||
| return type(self)(new_values, tz=self.tz, freq='infer') |
There was a problem hiding this comment.
Is the more complicated construction above necessary for some reason? AFAICT these are equivalent.
| values[self._isnan] = iNaT | ||
| return self._shallow_copy(values=values) | ||
|
|
||
| def _maybe_convert_timedelta(self, other): |
There was a problem hiding this comment.
Since #23031 this is no longer used by the Array class; moved back to the Index class.
2 participants
git diff upstream/master -u -- "*.py" | flake8 --diff