java - 在 Eclipse 中使用 Java 代码将数据插入到 csv 文件

Question

我正在尝试使用 Movielens 100k 电影数据集开发一个推荐系统。对于数据集中已经存在的用户 ID 来说，它工作得很好，但我想注册一个新用户，并在固定的编号上获得他的评分。电影(比如 5 部)，然后根据分析给他推荐。为了进行分析，我需要将他提供的评分(对 5 部电影)添加到我的 csv 形式的数据集中。从屏幕截图检查我的 eclipse maven 目录结构。我想修改数据文件夹下的 ratings.csv。

一旦我从前端获取数据，如何将其附加到 ratings.csv 中以扩展我的数据集，从而允许新用户获得推荐。

ratings.csv 包含以下列

  user_id     movie_id    ratings

我可以以某种方式将数据附加到 ratings.csv

Answer 1

private static void writeToCSV(List<Data> dataList) throws FileNotFoundException {
    PrintWriter pw = new PrintWriter(new File("D:\\test.csv"));
    StringBuilder sb = new StringBuilder();
    sb.append("PAN,");
    sb.append("REASON,");
    sb.append("DATE");
    sb.append("\n");

    dataList.stream().map((data) -> {
        sb.append(data.getPan() + ",");
        return data;
    }).map((data) -> {
        sb.append(data.getReason() + ",");
        return data;
    }).map((data) -> {
        sb.append(data.getDateTime() + "");
        return data;
    }).forEach((_item) -> {
        sb.append("\n");
    });

    pw.write(sb.toString());
    pw.close();
    System.out.println("done!");
}

如果你想用空格格式化 csv 数据，那么你可以使用 StringUtils (org.apache.commons.lang)

private static String rightPad(String text, int length) {
        return StringUtils.rightPad(text, length, " ");
    }

    private static String leftPad(String text, int length) {
        return StringUtils.leftPad(text, length, " ");
    }

因此，要将数据写入 csv 文件，我认为不需要另一个库