java - Spark : createDataset() into a Dataset?

Question

这个问题已经有答案了:

Dataframe from List<String> in Java (3 个回答)

已关闭 4 年前。

由于API原因，我的测试程序需要得到Dataset<Row>设置测试，但我找不到任何更简单的方法来使用 createDataset() 进行测试比这个:

Dataset<String> datasetString =
    spark.createDataset(Arrays.asList(
        "Country",
        "UK", "US", "UK", "US", "UK",
        "US", "UK", "US", "UK", "US")
        , Encoders.STRING());
Dataset<Row> dfOneCol =
    spark
        .read()
        .format("csv")
        .option("header", "true")
        .csv(datasetString);

有没有更简单的方法？

Answer 1

试试这个:

List<String> points = Arrays.asList("UK", "US", "Foo", "Bar");
dataset = spark.createDataset(points, Encoders. STRING()).toDF("Country");

希望对你有帮助