Java操作CSV [vaynexiao]

csv简介

csv只需要建一个txt，输入以下内容，后缀改为csv即可：
姓名,年龄,身高,ID,地址
UZI,23,177,RNG-uzi,中国
Faker,19,178,hide on bush,韩国

或者新建一个excel文件，输入以上内容，另存为csv类型。

csv与excel文件区别就是，前者文本文件，后者二进制文件

javacsv

		<dependency>
			<groupId>net.sourceforge.javacsv</groupId>
			<artifactId>javacsv</artifactId>
			<version>2.0</version>
		</dependency>

import com.csvreader.CsvReader;
import com.csvreader.CsvWriter;
import org.junit.jupiter.api.Test;
import org.springframework.boot.test.context.SpringBootTest;

import java.io.IOException;
import java.nio.charset.Charset;
import java.util.ArrayList;
import java.util.List;

@SpringBootTest
class CSV2 {

    @Test
    public void testWriteAndRead(){
        /*
            dept_code,dept_name,dept_code_2
            dept000,p0,RSA000
            dept001,p1,RSA001
            0000000,StemQ,code2_0
            dept002,p2,RSA002
            1000000,StemQ,code2_1
        */
        // 数据每次会覆盖
        String filePath = "javacsv.csv";

        // 创建CSV写对象
        CsvWriter csvWriter = new CsvWriter(filePath,',', Charset.forName("UTF-8"));

        String[] headers = {"dept_code","dept_name","dept_code_2"};
        // 一行数据为一个String[]
        String[] content = {"dept000","p0","RSA000"};
        List<String[]> records = new ArrayList<>();
        records.add(new String[]{"dept001", "p1" ,"RSA001"});
        records.add(new String[]{"dept002" , "p2" , "RSA002"});
        try {
            csvWriter.writeRecord(headers); // 写表头
            csvWriter.writeRecord(content); // 写入单条数据
            for (int i = 0; i < records.size(); i++) { // 写入多条数据
                csvWriter.writeRecord(records.get(i));
                String[] csvContent = { i + "000000", "StemQ", "code2_" + i };
                csvWriter.writeRecord(csvContent);
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
        csvWriter.close();

        CsvReader csvReader = null;
        try {
            // 创建CSV读对象
            csvReader = new CsvReader(filePath);
            // 跳过表头 如果需要表头的话，这句可以忽略
            csvReader.readHeaders();
            while (csvReader.readRecord()){
                System.out.println(csvReader.getRawRecord()); // 读一整行
                System.out.println(csvReader.get(0) + ":" + csvReader.get("dept_code_2")); // 读这行的某一列
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
        csvReader.close();
    }
}

打印如下：
dept000,p0,RSA000
dept000:RSA000
dept001,p1,RSA001
dept001:RSA001
0000000,StemQ,code2_0
0000000:code2_0
dept002,p2,RSA002
dept002:RSA002
1000000,StemQ,code2_1
1000000:code2_1

commons-csv

		<dependency>
			<groupId>org.apache.commons</groupId>
			<artifactId>commons-csv</artifactId>
			<version>1.8</version>
		</dependency>

import org.apache.commons.csv.CSVRecord;
import org.junit.jupiter.api.Test;
import org.springframework.boot.test.context.SpringBootTest;

import java.io.*;
import java.util.ArrayList;
import java.util.List;
import org.apache.commons.csv.CSVFormat;
import org.apache.commons.csv.CSVParser;
import org.apache.commons.csv.CSVPrinter;

@SpringBootTest
class CSV3 {

    @Test
    public void TestWrite() throws Exception {
        /*
            "dept001,产品部,RSA001"
            "dept002,设计部,RSA002"
            "dept003,技术部,RSA003"
            "dept004,运营部,RSA004"
            "dept005,财务部,RSA005"
        */
        String[] header = new String[]{"dept_code", "dept_name", "leader_code"};
        CSVFormat csvFormat = CSVFormat.DEFAULT
                .withHeader(header)
                .withSkipHeaderRecord()
                .withDelimiter(',');
        Appendable out = new PrintWriter("commons-csv.csv");

        //CSVFormat csvFormat = CSVFormat.newFormat(',').withHeader(header).withSkipHeaderRecord();
        // 注意 使用 DEFAULT 创建的CSVFormat对象生成记录可以自动换行,
        // 如果使用上面的 newFormat 的csvformat 不会自动换行
        //  CSVPrinter csvPrinter = CSVFormat.DEFAULT.print(out);
        CSVPrinter csvPrinter = new CSVPrinter(out, csvFormat);

        List<String> records = new ArrayList<>();
        records.add("dept001" + "," + "产品部" + "," + "RSA001");
        records.add("dept002" + "," + "设计部" + "," + "RSA002");
        records.add("dept003" + "," + "技术部" + "," + "RSA003");
        records.add("dept004" + "," + "运营部" + "," + "RSA004");
        records.add("dept005" + "," + "财务部" + "," + "RSA005");
        for (String record : records) {
            csvPrinter.printRecord(record);
        }
        csvPrinter.flush();
        csvPrinter.close();

    }

    @Test
    public void testRead() throws Exception {
        String[] header = new String[]{"dept_code", "dept_name", "leader_code"};
        String FILE_NAME = "commons-csv.csv";
        // 注意 : withSkipHeaderRecord 这里要设置跳过 header, 否则会吧 header 当成第一行记录
        CSVFormat csvFormat = CSVFormat.DEFAULT
                .withHeader(header)
                .withSkipHeaderRecord()
                // 如果默认分隔符不是，则需要手动设置
                .withDelimiter(',');
        Reader in = new FileReader(FILE_NAME);
        CSVParser parser = csvFormat.parse(in);
        List<CSVRecord> records = parser.getRecords();
        for (CSVRecord record : records) {
            String dept_code = record.get("dept_code");
            //String dept_name = record.get("dept_name"); ???为什么这句就报错
            //String leader_code = record.get("leader_code");
            //System.out.println(dept_code+":"+dept_name+":"+leader_code);
            System.out.println(dept_code);
            System.out.println(record.get(0));
            System.out.println(record.toString());
        }
        parser.close();
        in.close();
    }
}