将输出格式化为CSV文件格式以考虑某些行中的空值

时间:2015-10-12 18:56:00

标签: java csv file-io hashmap bufferedreader

tl; dr:我想以我正在阅读的相同CSV文件格式输出我的HashMap

我想以此为前言:我可能会以错误的方式解决这个问题。因为我开始思考一件事,并没有解释我遇到的这个问题。因此,我尝试制作一个小型应用程序,根据您的心情类型为您提供随机观看的电影(类似于Netflix的Max应用程序,但相当愚蠢)。我有一个电影列表,我将以CSV格式格式化自己,因为我最近编写了一些从CSV文件中读取值的代码,但我没有太多要改变的地方。

这就是困境:我已经读过CSV格式的文件(只有两行示例文件),因为我知道哪些列包含了我使用BufferedReader来逐行读取存储每个值的行它自己ArrayList中的分隔值(我知道有更好的方法,但这就是我现在提出的方法)然后我根据类型将每个ArrayList存储到HashMap。现在我希望能够在某个时刻写回同一个文件来编辑它。因此,已经观看的电影将从HashMap中删除,然后覆盖该文件,以便在下次回读时,已经观看的电影将不再存在于该文件中。因此,此时我遇到的困难是将输出格式化为实际CSV中的空白区域。

因此,例如,测试文件我只包含两行,每个电影类型有两部电影,除了戏剧和喜剧。所以文件看起来像这样

Action,Drama,Sci-fi/Fantasy,Thriller/Suspense,Comedy Action2,,Sci-fi/Fantasy2,Thriller/Suspense2,

只是为了巩固我想要/期望的输出;说我看科幻/幻想2,这是一部好电影,但它已经去了,我想要删除的输出就是这个

Action,Drama,Sci-fi/Fantasy,Thriller/Suspense,Comedy Action2,,,Thriller/Suspense2,

但是我知道我不会得到那些结果,因为当我只是读取文件然后输出它我得到:

Action,Action2, Drama,, Thriller/Suspense,Thriller/Suspense2, Comedy,, Sci-fi/Fantasy,Sci-fi/Fantasy2,

所以在得到这些结果后,我现在意识到我没有做好足够的计划,并想知道我是否采取了错误的方式。有谁知道如何格式化我描述的方式?我试图找到一个解决方案,但在空手而归后,我推断出我想要的格式可能与CSV的外观有关,考虑到文件中的某些单元格必须是空白的。我尝试保留文件中的任何空格,以便它们进入hashmap,但结果是相同的。我想过完全绕过输出文件,但我不确定如何保存我最初阅读的值进入我的地图。任何想法或解决方案将非常感谢。这是完成所有工作的课程:

 package rngesus;

 import java.io.BufferedReader;
 import java.io.File;
 import java.io.FileNotFoundException;
 import java.io.FileReader;
 import java.io.FileWriter;
 import java.io.IOException;
 import java.util.ArrayList;
 import java.util.HashMap;

 public class ReadWrite {

private static final String[] GENRES = { "Action", "Drama",
        "Sci-Fi/Fantasy", "Thriller/Suspense", "Comedy" };
private static final String NEW_LINE = "\n";
private static final String DELIMITER = ",";
private static final int NUM_OF_COL = 5;
private static final int GENRE_1 = 0;
private static final int GENRE_2 = 1;
private static final int GENRE_3 = 2;
private static final int GENRE_4 = 3;
private static final int GENRE_5 = 4;

private static String moviesFile;
private HashMap<String, ArrayList<String>> moviesByGenre;

ArrayList<String> actionMovies;
ArrayList<String> dramaMovies;
ArrayList<String> sciFiFantasyMovies;
ArrayList<String> thrillerSuspenseMovies;
ArrayList<String> comedyMovies;

public ReadWrite() {
    moviesFile = "";
    moviesByGenre = new HashMap<String, ArrayList<String>>();
    actionMovies = new ArrayList<String>();
    dramaMovies = new ArrayList<String>();
    sciFiFantasyMovies = new ArrayList<String>();
    thrillerSuspenseMovies = new ArrayList<String>();
    comedyMovies = new ArrayList<String>();
}

public void readAndSortInputFile(String fileOfMovies) throws IOException {

    try {
        BufferedReader buffRdr = new BufferedReader(new FileReader(
                new File(fileOfMovies)));
        String line = "";
        while ((line = buffRdr.readLine()) != null) {
            String[] lnPtr = line.split(",", NUM_OF_COL);
            int diff = Math.min(lnPtr.length, NUM_OF_COL);
            for (int i = 0; i < diff; i++) {
                if ((i == GENRE_1) && !lnPtr[i].isEmpty()) {
                    actionMovies.add(lnPtr[i]);
                } else if ((i == GENRE_2) && !lnPtr[i].isEmpty()) {
                    dramaMovies.add(lnPtr[i]);
                } else if ((i == GENRE_3) && !lnPtr[i].isEmpty()) {
                    sciFiFantasyMovies.add(lnPtr[i]);
                } else if ((i == GENRE_4) && !lnPtr[i].isEmpty()) {
                    thrillerSuspenseMovies.add(lnPtr[i]);
                } else if ((i == GENRE_5) && !lnPtr[i].isEmpty()){
                    comedyMovies.add(lnPtr[i]);
                }
            }
        }
        buffRdr.close();
        moviesFile = fileOfMovies;
    } catch (FileNotFoundException err) {
        err.printStackTrace();
        System.out.println("Error: Unable to locate file specified");
    }
}

public void mapMoviesToGenre(){
    moviesByGenre.put(GENRES[GENRE_1], actionMovies);
    moviesByGenre.put(GENRES[GENRE_2], dramaMovies);
    moviesByGenre.put(GENRES[GENRE_3], sciFiFantasyMovies);
    moviesByGenre.put(GENRES[GENRE_4], thrillerSuspenseMovies);
    moviesByGenre.put(GENRES[GENRE_5], comedyMovies);

}

public void initMapToOutput() {

    mapMoviesToGenre();
    FileWriter fileWriter = null;
    try {
        fileWriter = new FileWriter(moviesFile);
        fileWriter.append(NEW_LINE);

        for(ArrayList<String> moviesInGenre : moviesByGenre.values()){
            for(String movie : moviesInGenre){
                fileWriter.append(movie);
                fileWriter.append(DELIMITER);
            }
            fileWriter.append(NEW_LINE);
        }

    } catch (Exception err) {
        err.printStackTrace();

    } finally {
        try {
            fileWriter.flush();
            fileWriter.close();
        } catch (IOException err) {
            err.printStackTrace();
        }
    }

}

}

我不认为格式很好,但我对此有些无知,对任何人都可以随意编辑。我已经意识到必须有一种更好的方法来存储从文件中读取的值,但是现在我专注于解决另一个难题。但如果有人愿意指出任何其他的东西,请随意。

1 个答案:

答案 0 :(得分:0)

好的,所以我通过它来解决它可能关心的问题。我继续前进并逐行分组电影,而不是逐列。我所要做的就是将代码的while部分更改为

 int index = 0;
        while ((line = buffRdr.readLine()) != null) {
            String[] lnPtr = line.split(",", NUM_OF_COL);
            int diff = Math.min(lnPtr.length, NUM_OF_COL);
            for (int i = 0; i < diff; i++) {
                if ((index == GENRE_1) && !lnPtr[i].isEmpty()) {
                    actionMovies.add(lnPtr[i]);
                } else if ((index == GENRE_2) && !lnPtr[i].isEmpty()) {
                    dramaMovies.add(lnPtr[i]);
                } else if ((index == GENRE_3) && !lnPtr[i].isEmpty()) {
                    sciFiFantasyMovies.add(lnPtr[i]);
                } else if ((index == GENRE_4) && !lnPtr[i].isEmpty()) {
                    thrillerSuspenseMovies.add(lnPtr[i]);
                } else if ((index == GENRE_5) && !lnPtr[i].isEmpty()){
                    comedyMovies.add(lnPtr[i]);
                }
            }
            index++;
        }

我现在每次都在同一行上播放相同类型的电影,虽然有时候会有不同的顺序,我认为是由于进出HashMap。如果有人知道更清洁更有效的方法,请随时分享,但我现在解决了我的问题。