Update cols_to_array to accept *cols, rather than only [a, b, c, ...]

Naissant / dendri

Common Healthcare feature engineering algorithms implemented in PySpark.

MIT License

2 stars 1 forks source link

Update cols_to_array to accept *cols, rather than only [a, b, c, ...] #2

Closed WesRoach closed 3 years ago

WesRoach commented 3 years ago

dataframe.cols_to_array currently only accepts a list of columns. To fall in line better with the PySpark API the function should accept *args of columns.

Accept:

List[str]
List[Column]
*cols of Column
*cols of str