Skip to content

Commit 52bb549

Browse files
committed
Merge branch 'iterable-names-in-select' into col
2 parents c3c89cd + 0ef4852 commit 52bb549

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

spec/API_specification/dataframe_api/dataframe_object.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -109,13 +109,13 @@ def col(self, name: str, /) -> Column:
109109
"""
110110
...
111111

112-
def select(self, names: Sequence[str], /) -> Self:
112+
def select(self, *names: str) -> Self:
113113
"""
114114
Select multiple columns by name.
115115
116116
Parameters
117117
----------
118-
names : Sequence[str]
118+
*names : str
119119
120120
Returns
121121
-------

spec/API_specification/examples/tpch/q5.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ def query(
6464
result.col("l_extendedprice") * (1 - result.col("l_discount"))
6565
).rename("revenue")
6666
result = result.assign(new_column)
67-
result = result.select(["revenue", "n_name"])
67+
result = result.select("revenue", "n_name")
6868
result = result.group_by("n_name").sum()
6969

7070
return result.dataframe

0 commit comments

Comments
 (0)