2012-06-26 86 views
5

我正在处理csv解析器,我想分别读取头文件和csv文件的其余部分。 这是我的代码来阅读csv。csv解析器读取头文件

当前代码读取csv文件中的所有内容,但我需要单独阅读标题。 请帮我解决这个问题。

public class csv { 

private void csvRead(File file) 
{ 
    try 
    { 
    BufferedReader br = new BufferedReader(new FileReader(file)); 
    String strLine = ""; 
    StringTokenizer st = null; 
    File cfile=new File("csv.txt"); 
    BufferedWriter writer = new BufferedWriter(new FileWriter(cfile)); 
    int tokenNumber = 0; 

    while((strLine = br.readLine()) != null) 
    { 
      st = new StringTokenizer(strLine, ","); 
      while(st.hasMoreTokens()) 
      { 

        tokenNumber++; 
        writer.write(tokenNumber+" "+ st.nextToken()); 
        writer.newLine(); 
      } 


      tokenNumber = 0; 
      writer.flush(); 
    } 
} 

    catch(Exception e) 
    { 
     e.getMessage(); 
    } 
} 
+1

如果您想要将文件头的读取与文件的其余部分分开,则需要一个只处理第一行的方法,另一个方法处理文件中的所有其他行。 –

回答

3

你认为是OpenCSV

这里前一个问题...

CSV API for Java

貌似可以很容易打出头...

String fileName = "data.csv"; 
CSVReader reader = new CSVReader(new FileReader(fileName)); 


// if the first line is the header 
String[] header = reader.readNext(); 

// iterate over reader.readNext until it returns null 
String[] line = reader.readNext(); 
5

请考虑使用Commons CSV。该库根据RFC 4180 - Common Format and MIME Type for Comma-Separated Values (CSV) Files编写。什么是兼容的读取这样的台词:

"aa,a","b""bb","ccc" 

而且使用相当简单,仅存在3类,并根据小样本文档:

有标签为CSV串的解析分离器,“'作为一个可选的 值封装和评论以‘#’:

CSVFormat format = new CSVFormat('\t', '"', '#'); 
Reader in = new StringReader("a\tb\nc\td"); 
String[][] records = new CSVParser(in, format).getRecords(); 

,另外你得到这个解析器已经作为常量:

  • DEFAULT - 如RFC 4180
  • EXCEL定义标准逗号分隔格式 - Excel文件格式(使用逗号作为值的分隔符)。
  • MYSQL - SELECT INTO OUTFILE和LOAD DATA INFILE操作使用的默认MySQL格式。 TDF - 制表分隔格式。
3

我们在CSVFormat中提供了withHeader()方法。如果你使用这个选项,那么你将能够使用头文件读取文件。

CSVFormat format = CSVFormat.newFormat(',').withHeader(); 
Map<String, Integer> headerMap = dataCSVParser.getHeaderMap(); 

会给你所有的标题。

public class CSVFileReaderEx { 
    public static void main(String[] args){ 
     readFile(); 
    } 

    public static void readFile(){ 
     List<Map<String, String>> csvInputList = new CopyOnWriteArrayList<>(); 
     List<Map<String, Integer>> headerList = new CopyOnWriteArrayList<>(); 

     String fileName = "C:/test.csv"; 
     CSVFormat format = CSVFormat.newFormat(',').withHeader(); 

      try (BufferedReader inputReader = new BufferedReader(new FileReader(new File(fileName))); 
        CSVParser dataCSVParser = new CSVParser(inputReader, format);) { 

      List<CSVRecord> csvRecords = dataCSVParser.getRecords(); 

      Map<String, Integer> headerMap = dataCSVParser.getHeaderMap(); 
       headerList.add(headerMap); 
       headerList.forEach(System.out::println); 

      for(CSVRecord record : csvRecords){ 
       Map<String, String> inputMap = new LinkedHashMap<>(); 

       for(Map.Entry<String, Integer> header : headerMap.entrySet()){ 
        inputMap.put(header.getKey(), record.get(header.getValue())); 
       } 

       if (!inputMap.isEmpty()) { 
        csvInputList.add(inputMap); 
       } 
      } 

      csvInputList.forEach(System.out::println); 

      } catch (Exception e) { 
      System.out.println(e); 
      } 
    } 
} 
+0

不错,真的很不错 –