Skip to content

Commit 78a26a0

Browse files
jbrockmendelSeeminSyed
authored andcommitted
ENH: implement EA.size (pandas-dev#32644)
1 parent d46cd83 commit 78a26a0

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
@@ -407,6 +407,13 @@ def shape(self) -> Tuple[int, ...]:
407407
"""
408408
return (len(self),)
409409

410+
@property
411+
def size(self) -> int:
412+
"""
413+
The number of elements in the array.
414+
"""
415+
return np.prod(self.shape)
416+
410417
@property
411418
def ndim(self) -> int:
412419
"""

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)