Bucketizer pyspark

x = [(0,18.0),(1,19.0),(2,8.0),(3,5.0),(4,2.2),(5,4.0)]
d = spark.createDataFrame(x,["id","hour"])
d.show()
from pyspark.ml.feature import Bucketizer
splits = [0,1,2,3,4,float("Inf")]
buck = Bucketizer(splits=splits,inputCol="BATHSTOTAL",outputCol="baths")
df = buck.transform(df)
df[["BATHSTOTAL","baths"]].show()

Similar codes

0 vote316 views1 answer
Z
Python
4 votes319 views1 answer
Tkinter frame example
Python
184 votes391 views6 answers
Update pip python
Python
0 vote401 views1 answer
Pandas index from 1
Python
0 vote307 views1 answer
Remove percentage in python
Python
0 vote360 views1 answer
Get or create django
Python Django
21 votes324 views6 answers
Remove columns from dataframe
Python
4 votes328 views2 answers
How to get the most common number in python
Python
1 vote298 views1 answer
Decrease hours in datetime python
Python
3 votes295 views3 answers
Python iterar claves
Python