擅长:python、mysql、java
<p>看看这个文件:<a href="https://spark.apache.org/docs/2.1.0/ml-features.html#vectorassembler" rel="nofollow noreferrer">https://spark.apache.org/docs/2.1.0/ml-features.html#vectorassembler</a></p>
<pre><code>from pyspark.ml.linalg import Vectors
from pyspark.ml.feature import VectorAssembler
assembler = VectorAssembler(
inputCols=["mark1", "mark2", "mark3"],
outputCol="marks")
output = assembler.transform(dataset)
output.select("name", "marks").show(truncate=False)
</code></pre>