Skip to content

Commit d2e8549

Browse files
jbrockmendelsimonjayhawkins
authored andcommitted
ENH: implement EA.size (pandas-dev#32644)
1 parent 7719aaa commit d2e8549

File tree

2 files changed

+10
-0
lines changed

2 files changed

+10
-0
lines changed

pandas/core/arrays/base.py

+7
Original file line numberDiff line numberDiff line change
@@ -402,6 +402,13 @@ def shape(self) -> Tuple[int, ...]:
402402
"""
403403
return (len(self),)
404404

405+
@property
406+
def size(self) -> int:
407+
"""
408+
The number of elements in the array.
409+
"""
410+
return np.prod(self.shape)
411+
405412
@property
406413
def ndim(self) -> int:
407414
"""

pandas/tests/extension/base/interface.py

+3
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,9 @@ class BaseInterfaceTests(BaseExtensionTests):
1919
def test_len(self, data):
2020
assert len(data) == 100
2121

22+
def test_size(self, data):
23+
assert data.size == 100
24+
2225
def test_ndim(self, data):
2326
assert data.ndim == 1
2427

0 commit comments

Comments
 (0)