iterable argument in select (data-apis#300)

Author: MarcoGorelli

spec/API_specification/dataframe_api/dataframe_object.py

@@ -109,13 +109,13 @@ def get_column_by_name(self, name: str, /) -> Column:
         """
         ...
 
-    def select(self, names: Sequence[str], /) -> Self:
+    def select(self, *names: str) -> Self:
         """
         Select multiple columns by name.
 
         Parameters
         ----------
-        names : Sequence[str]
+        *names : str
 
         Returns
         -------

spec/API_specification/examples/tpch/q5.py

@@ -68,7 +68,7 @@ def query(
         * (1 - result.get_column_by_name("l_discount"))
     ).rename("revenue")
     result = result.assign(new_column)
-    result = result.select(["revenue", "n_name"])
+    result = result.select("revenue", "n_name")
     result = result.group_by("n_name").sum()
 
     return result.dataframe