将csv文件读入HashMap <string,arraylist <integer =“”>&gt;

时间:2016-07-17 15:16:12

标签: java csv arraylist hashmap

我一直在尝试制作一个java程序,其中逐行读取制表符分隔的csv文件,并将第一列(这是一个字符串)作为键添加到哈希映射和第二列(整数)是它的值。

在输入文件中,有重复的键,但具有不同的值,因此我打算将值添加到现有键以形成值的ArrayList。

我无法弄清楚这样做的最佳方式,并且想知道是否有人可以提供帮助?

由于

编辑:对不起伙计们,到目前为止我已经使用了代码: 我应该添加第一列是值,第二列是键。

public class WordNet {

    private final HashMap<String, ArrayList<Integer>> words;
    private final static String LEXICAL_UNITS_FILE = "wordnet_data/wn_s.csv";

    public WordNet() throws FileNotFoundException, IOException {

        words = new HashMap<>();
        readLexicalUnitsFile();
    }

    private void readLexicalUnitsFile() throws FileNotFoundException, IOException{

        BufferedReader in = new BufferedReader(new FileReader(LEXICAL_UNITS_FILE));
        String line;

        while ((line = in.readLine()) != null) {
            String columns[] = line.split("\t");
            if (!words.containsKey(columns[1])) {
                words.put(columns[1], new ArrayList<>());
            }

        }
        in.close();

    }

3 个答案:

答案 0 :(得分:2)

你很亲密

String columns[] = line.split("\t");
if (!words.containsKey(columns[1])) {
    words.put(columns[1], new ArrayList<>());
}

应该是

String columns[] = line.split("\t");
String key = columns[0];                // enhance readability of code below
List<Integer> list = words.get(key);    // try to fetch the list
if (list == null)                       // check if the key is defined
{                                       //   if not
    list = new ArrayList<>();           //      create a new list
    words.put(key,list);                //      and add it to the map
}
list.add(new Integer(columns[1]));      // in either case, add the value to the list

回应OP的评论/问题

  

...最后一行只是将整数添加到列表中而不是添加到hashmap中,之后是否需要添加某些内容?

声明后

List<Integer> list = words.get(key);

有两种可能性。如果list为非null,则它是引用到(不是副本)已在地图中的列表。

如果listnull,那么我们知道地图不包含给定的密钥。在这种情况下,我们创建一个新的空列表,将变量list设置为对新创建的列表的引用,然后将列表添加到键的映射中。

在任何一种情况下,当我们达到

list.add(new Integer(columns[1]));

变量list包含对地图中已有的ArrayList的引用,可以是之前存在的引用,也可以是我们刚创建和添加的引用。我们只是为它添加值。

答案 1 :(得分:0)

  

我应该添加第一列是值,第二列是键。

您可以通过List声明重新声明ArrayList声明。但这不是很成问题。

无论如何,没有经过测试,但逻辑应该如下:

    while ((line = in.readLine()) != null) {
      String columns[] = line.split("\t");
      ArrayList<Integer> valueForCurrentLine = words.get(columns[1]);

      // you instantiate and put the arrayList once
      if (valueForCurrentLine==null){
          valueForCurrentLine = new  ArrayList<Integer>();
          words.put(columns[1],valueForCurrentLine);
      }

      valueForCurrentLine.add(columns[0]);

答案 2 :(得分:0)

赞成Jim Garrison的回答。这里多了一点......(是的,你应该检查/标记他的答案是解决它的答案)

import java.io.BufferedReader;
import java.io.FileNotFoundException;
import java.io.FileReader;
import java.io.IOException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class WordNet {

    private final Map<String, List<Integer>> words;
    private final static String LEXICAL_UNITS_FILE = "src/net/bwillard/practice/code/wn_s.csv";

    /**
     * 
     * @throws FileNotFoundException
     * @throws IOException
     */
    public WordNet() throws FileNotFoundException, IOException {
        words = new HashMap<>();
        readLexicalUnitsFile();
    }

    /**
     * 
     * @throws FileNotFoundException
     * @throws IOException
     */
    private void readLexicalUnitsFile() throws FileNotFoundException, IOException {

        BufferedReader in = new BufferedReader(new FileReader(LEXICAL_UNITS_FILE));
        String line;

        while ((line = in.readLine()) != null) {
            String columns[] = line.split("\t");
            String key = columns[0];
            int valueInt;
            List<Integer> valueList;

            try {
                valueInt = Integer.parseInt(columns[1]);
            } catch (NumberFormatException e) {
                System.out.println(e);
                continue;
            }

            if (words.containsKey(key)) {
                valueList = words.get(key);
            } else {
                valueList = new ArrayList<>();
                words.put(key, valueList);
            }

            valueList.add(valueInt);
        }

        in.close();
    }

    //You can test this file by running it as a standalone app....
    public static void main(String[] args) {
        try {
            WordNet wn = new WordNet();
            for (String k : wn.words.keySet()) {
                System.out.println(k + " " + wn.words.get(k));
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}