Skip to content

Commit 1a27d2f

Browse files
committed
Add IntegerArray.size, fix pandas-dev#25580.
Signed-off-by: HE, Tao <[email protected]>
1 parent 74a9ae3 commit 1a27d2f

File tree

2 files changed

+14
-0
lines changed

2 files changed

+14
-0
lines changed

pandas/core/arrays/integer.py

+4
Original file line numberDiff line numberDiff line change
@@ -400,6 +400,10 @@ def __setitem__(self, key, value):
400400
def __len__(self):
401401
return len(self._data)
402402

403+
@property
404+
def size(self):
405+
return self._data.size
406+
403407
@property
404408
def nbytes(self):
405409
return self._data.nbytes + self._mask.nbytes

pandas/tests/resample/test_datetime_index.py

+10
Original file line numberDiff line numberDiff line change
@@ -100,6 +100,16 @@ def test_resample_basic(series, closed, expected):
100100
result = s.resample('5min', closed=closed, label='right').mean()
101101
assert_series_equal(result, expected)
102102

103+
def test_resample_integerarray():
104+
# resample on IntegerArray
105+
ts = pd.Series(range(9),
106+
index=pd.date_range('1/1/2000', periods=9, freq='T'),
107+
dtype='Int64')
108+
result = ts.resample('3T').sum()
109+
expected = Series([3, 12, 21],
110+
index=pd.date_range('1/1/2000', periods=3, freq='3T'),
111+
dtype="Int64")
112+
assert_series_equal(result, expected)
103113

104114
def test_resample_basic_grouper(series):
105115
s = series

0 commit comments

Comments
 (0)