public static void main(String[] args) throws IOException { Reader r = new Reader(); ArrayList<String> labels = r.read("test.label"); ArrayList<String> data = r.read("test.data"); FileWriter fw = new FileWriter("test20.txt"); for (int i = 1; i <= labels.size(); i++) { int label = Integer.parseInt(labels.get(i - 1)); String doc = "" + (label - 1); while (!data.isEmpty()) { String s = data.get(0); String[] temp = s.split(" "); if (Integer.parseInt(temp[0]) != i) { break; } data.remove(0); String oneWord = " " + temp[1] + ":" + temp[2]; doc = doc + oneWord; } doc = doc + "\r\n"; fw.write(doc); System.out.println("doc " + i + " finished"); } fw.close(); }
public ArrayList<OneDoc> readText(String fileName) { ArrayList<OneDoc> result = new ArrayList<OneDoc>(); Reader r = new Reader(); ArrayList<String> stringList = r.read(fileName); for (String s : stringList) { result.add(new OneDoc(s)); } return result; }