Skip to content

Commit d112e73

Browse files
committed
BUG: resampling with NaT in TimedeltaIndex (pandas-dev#13223)
1 parent 06b35db commit d112e73

File tree

2 files changed

+20
-2
lines changed

2 files changed

+20
-2
lines changed

pandas/tseries/resample.py

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1194,8 +1194,10 @@ def _get_time_delta_bins(self, ax):
11941194
data=[], freq=self.freq, name=ax.name)
11951195
return binner, [], labels
11961196

1197-
start = ax[0]
1198-
end = ax[-1]
1197+
# Addresses GH #13223
1198+
start = ax.min()
1199+
end = ax.max()
1200+
11991201
labels = binner = TimedeltaIndex(start=start,
12001202
end=end,
12011203
freq=self.freq,
@@ -1204,6 +1206,13 @@ def _get_time_delta_bins(self, ax):
12041206
end_stamps = labels + 1
12051207
bins = ax.searchsorted(end_stamps, side='left')
12061208

1209+
if ax.hasnans:
1210+
binner = binner.insert(0, tslib.NaT)
1211+
labels = labels.insert(0, tslib.NaT)
1212+
1213+
n_NaT = sum([ax_i is tslib.NaT for ax_i in ax])
1214+
bins = np.insert(bins, 0, n_NaT)
1215+
12071216
# Addresses GH #10530
12081217
if self.base > 0:
12091218
labels += type(self.freq)(self.base)

pandas/tseries/tests/test_resample.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -970,6 +970,15 @@ def test_resample_timedelta_idempotency(self):
970970
expected = series
971971
assert_series_equal(result, expected)
972972

973+
def test_resample_timedelta_missing_values(self):
974+
# GH 13223
975+
index = pd.to_timedelta(['0s', pd.NaT, '2s'])
976+
series = pd.Series([2, 3, 5], index=index)
977+
result = series.resample('1s').mean()
978+
expected = pd.Series([2, np.nan, 5], index=pd.timedelta_range(
979+
start='0s', end='2s', freq='1s'))
980+
assert_series_equal(result, expected)
981+
973982
def test_resample_rounding(self):
974983
# GH 8371
975984
# odd results when rounding is needed

0 commit comments

Comments
 (0)