commons-csv的基本操作

来源：互联网发布：数据分析工程师笔试题编辑：程序博客网时间：2024/06/05 01:10

本文主要介绍一些关系CSV的基本操作。

commons-csv 由两大核心对象组成 CSVParser（解析），CSVPrinter（写csv）下面附上工具类，以及测试类

package com.csv.demo;import java.io.FileInputStream;import java.io.FileNotFoundException;import java.io.FileOutputStream;import java.io.IOException;import java.io.InputStreamReader;import java.io.OutputStreamWriter;import org.apache.commons.csv.CSVFormat;import org.apache.commons.csv.CSVParser;import org.apache.commons.csv.CSVPrinter;public class CSVUtils{public static CSVParser getCSVParser(String filePath) throws IOException{CSVFormat format = CSVFormat.DEFAULT.withHeader();InputStreamReader isr = new InputStreamReader(new FileInputStream(filePath), "UTF-8");return new CSVParser(isr, format);}public static CSVPrinter getCSVPrinter(String filePath) throws IOException{CSVFormat format = CSVFormat.DEFAULT.withRecordSeparator("\n");OutputStreamWriter osw = new OutputStreamWriter(new FileOutputStream(filePath), "UTF-8");return new CSVPrinter(osw, format);}}

测试类：演示将一个CSV写到另一个CSV

package com.csv.demo;import java.io.IOException;import java.util.Iterator;import org.apache.commons.csv.CSVParser;import org.apache.commons.csv.CSVPrinter;import org.apache.commons.csv.CSVRecord;import org.apache.commons.io.IOUtils;public class Test{public static void main(String[] args) throws IOException{CSVParser parser = CSVUtils.getCSVParser("D://demo01.csv");CSVPrinter printer=CSVUtils.getCSVPrinter("D://demo02.csv");Iterator<CSVRecord> iterator = parser.iterator();printer.printRecord(parser.getHeaderMap().keySet());//写CSV第一行while(iterator.hasNext()){printer.printRecord(iterator.next());}IOUtils.closeQuietly(parser);IOUtils.closeQuietly(printer);}}

CSVRecord有一个比较常用的方法get(String name)，可以选取CSV的某一列进行读写，只需指定首列字段名称即可。

0 0