StringIndexer pyspark

df_i = spark.createDataFrame([
    ("a", 2.0),
    ("c", 1.0),
    ("b", 2.0),    
    ("a", 1.0),
    ("a", 0.0),   
    ("c", 1.0)
], ["categoryIndex", "categoryIndex2"])
stringID = StringIndexer(inputCol="categoryIndex",outputCol="categoryIndex2")
indexer = stringID.fit(df_i)
df_i = indexer.transform(df_i)
df_i.show()

Similar codes

92 votes179 views6 answers
Python choose random element from list
Python
3 votes195 views1 answer
Clean punctuation from string python
Python
1 vote169 views1 answer
Rock paper scissors python
Python
6 votes211 views3 answers
Login system in django
Python Django
0 vote127 views1 answer
Geopy set proxy
Python
0 vote168 views1 answer
Scikit learn tree
Python
2 votes184 views2 answers
Pd.dataframe initial columns
Python
0 vote176 views1 answer
Write lines python with line breaks
Python
0 vote181 views2 answers
Python default dictonary
Python
3 votes164 views1 answer
Pandas create a column from index
Python