let __getitem__ take sequence of int

Author: MarcoGorelli

spec/API_specification/dataframe_api/column_object.py

@@ -1,14 +1,24 @@
-from typing import NoReturn
+from __future__ import annotations
+
+from typing import NoReturn, overload, Sequence
 
 class Column:
     def __len__(self) -> int:
         """
         Return the number of rows.
         """
 
-    def __getitem__(self, row: int) -> object:
+    @overload
+    def __getitem__(self, key: int) -> object:
+        ...
+
+    @overload
+    def __getitem__(self, key: Sequence[int]) -> Column:
+        ...
+
+    def __getitem__(self, key: int | Sequence[int]) -> object | Column:
         """
-        Get the element at row index `row`.
+        Get the element at row index `key`.
         """
 
     def __iter__(self) -> NoReturn: