SNOW-1924268: Add support for list values in Series.str.__getitem__ (#3130)

Author: sfc-gh-helmeleegy

CHANGELOG.md

@@ -10,6 +10,10 @@
 
 ### Snowpark pandas API Updates
 
+#### New Features
+
+- Added support for list values in `Series.str.__getitem__` (`Series.str[...]`).
+
 ## 1.29.0 (2025-03-05)
 
 ### Snowpark Python API Updates

docs/source/modin/supported/series_str_supported.rst

@@ -15,7 +15,9 @@ the method in the left column.
 | (Series.str)                |                                 |                                                    |
 +-----------------------------+---------------------------------+----------------------------------------------------+
 | ``__getitem__``             | P                               | ``N`` if the `key` parameter is set to a non-int   |
-|                             |                                 |  scalar value.                                     |
+|                             |                                 |  scalar value. Only string and list data values are|
+|                             |                                 |  supported. All column values must be of the same  |
+|                             |                                 |  type.                                             |
 +-----------------------------+---------------------------------+----------------------------------------------------+
 | ``capitalize``              | Y                               |                                                    |
 +-----------------------------+---------------------------------+----------------------------------------------------+

tests/integ/modin/series/test_str_accessor.py

@@ -250,6 +250,47 @@ def test_str___getitem___string_key():
         snow_ser.str["a"]
 
 
+@pytest.mark.parametrize(
+    "key",
+    [
+        None,
+        [1, 2],
+        (1, 2),
+        {1: "a", 2: "b"},
+        -100,
+        -2,
+        -1,
+        0,
+        1,
+        2,
+        100,
+        slice(None, None, None),
+        slice(0, -1, 1),
+        slice(-100, 100, 1),
+    ],
+)
+@sql_count_checker(query_count=1)
+def test_str___getitem___list(key):
+    native_ser = native_pd.Series([["a", "b"], ["c", "d", None], None, []])
+    snow_ser = pd.Series(native_ser)
+    eval_snowpark_pandas_result(
+        snow_ser,
+        native_ser,
+        lambda ser: ser.str[key],
+    )
+
+
+@sql_count_checker(query_count=0)
+def test_str___getitem___list_neg():
+    native_ser = native_pd.Series([["a", "b"], ["c", "d", None], None, []])
+    snow_ser = pd.Series(native_ser)
+    with pytest.raises(
+        NotImplementedError,
+        match="does not yet support 'step!=1' for list values",
+    ):
+        snow_ser.str[slice(None, None, 2)]
+
+
 @pytest.mark.parametrize(
     "pat",
     [