Skip to content

Commit c972dbc

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

File tree

2 files changed

+16
-0
lines changed

2 files changed

+16
-0
lines changed

pandas/core/arrays/integer.py

Lines changed: 4 additions & 0 deletions
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

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -101,6 +101,18 @@ def test_resample_basic(series, closed, expected):
101101
assert_series_equal(result, expected)
102102

103103

104+
def test_resample_integerarray():
105+
# resample on IntegerArray
106+
ts = pd.Series(range(9),
107+
index=pd.date_range('1/1/2000', periods=9, freq='T'),
108+
dtype='Int64')
109+
result = ts.resample('3T').sum()
110+
expected = Series([3, 12, 21],
111+
index=pd.date_range('1/1/2000', periods=3, freq='3T'),
112+
dtype="Int64")
113+
assert_series_equal(result, expected)
114+
115+
104116
def test_resample_basic_grouper(series):
105117
s = series
106118
result = s.resample('5Min').last()

0 commit comments

Comments
 (0)