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

2 votes263 views1 answer
Kaggle vs colab
Python
0 vote265 views1 answer
Sort eigenvalues and eigenvectors python
Python
1 vote307 views1 answer
Sort a list by length python
Python
0 vote286 views1 answer
Spread dictionary python
Python
0 vote261 views1 answer
How to print two values from a list in python
Python
8 votes279 views1 answer
Module to read keyboard
Python
0 vote259 views1 answer
Map dataframe parallel
Python
0 vote302 views2 answers
Welcome system
Python
0 vote253 views1 answer
Import data from website pandas python medium
Python
0 vote257 views1 answer
List out the groups from groupby
Python