Java 学习笔记

所有文章
📖 最新文章 transmittable-thread-local 库 JOOR 反射库 Lombok 库:为你减少样板代码 Slf4j 日志组件的使用 加速maven、gradle依赖下载
📖 Java 基础 安装 第一个程序 使用 UTF-8 编写代码 空值 null 正则表达式 线程 日期/时间 匿名类 枚举 ThreadLocal 线程本地变量 动态代理 jar 命令

使用 Java 生成 CSV 文件


示例1

package com.example;

import java.io.File;
import java.io.FileNotFoundException;
import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;
import java.util.stream.Stream;

public class TestCsv {

    /**
     * 转义字符串中的特殊字符,例如英文逗号
     * @param data
     * @return
     */
    public static String escapeSpecialCharacters(String data) {
        String escapedData = data.replaceAll("\\R", " ");
        if (data.contains(",") || data.contains("\"") || data.contains("'")) {
            data = data.replace("\"", "\"\"");
            escapedData = "\"" + data + "\"";
        }
        return escapedData;
    }

    /**
     * 将字符串数转换为csv格式的一行内容,用英文逗号分隔
     * @param data
     * @return
     */
    public static String convertToCSV(String[] data) {
        return Stream.of(data)
                .map(TestCsv::escapeSpecialCharacters)
                .collect(Collectors.joining(","));
    }

    /**
     * 写 csv 文件
     *
     * @param filePath
     * @param dataLineList
     */
    public static void writeCsvFile(String filePath, List<String[]> dataLineList) {
        File csvOutputFile = new File(filePath);
        try (PrintWriter pw = new PrintWriter(csvOutputFile)) {
            dataLineList.stream()
                    .map(TestCsv::convertToCSV)
                    .forEach(pw::println);
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        }
    }


    public static void main(String[] args) {
        List<String[]> dataLineList = new ArrayList<>();
        dataLineList.add(new String[]
                { "你好", "带换行\n和带英文逗号," });
        dataLineList.add(new String[]
                { "世界", "中国" });
        writeCsvFile("/tmp/test.csv", dataLineList);
    }

}

运行后,查看csv文件内容:

$ cat /tmp/test.csv
你好,"带换行
和带英文逗号,"
世界,中国

使用 WPS 打开文件:

示例2:指定文件编码

package com.example;

import java.io.*;
import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;
import java.util.stream.Stream;

public class TestCsv {

    /**
     * 转义字符串中的特殊字符,例如英文逗号
     * @param data
     * @return
     */
    public static String escapeSpecialCharacters(String data) {
        String escapedData = data.replaceAll("\\R", " ");
        if (data.contains(",") || data.contains("\"") || data.contains("'")) {
            data = data.replace("\"", "\"\"");
            escapedData = "\"" + data + "\"";
        }
        return escapedData;
    }

    /**
     * 将字符串数转换为csv格式的一行内容,用英文逗号分隔
     * @param data
     * @return
     */
    public static String convertToCSV(String[] data) {
        return Stream.of(data)
                .map(TestCsv::escapeSpecialCharacters)
                .collect(Collectors.joining(","));
    }

    /**
     * 写 csv 文件
     *
     * @param filePath
     * @param dataLineList
     * @param fileEncoding  文件编码,如 GBK
     */
    public static void writeCsvFile(String filePath, List<String[]> dataLineList, String fileEncoding) {
        try (BufferedWriter writer = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(filePath), fileEncoding))) {
            for (String[] dataLine : dataLineList) {
                String s = convertToCSV(dataLine);
                writer.write(s);
                writer.newLine();
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }


    public static void main(String[] args) {
        List<String[]> dataLineList = new ArrayList<>();
        dataLineList.add(new String[]
                { "你好", "带换行\n和带英文逗号," });
        dataLineList.add(new String[]
                { "世界", "中国" });
        writeCsvFile("/tmp/test1.csv", dataLineList, "GBK");
    }

}

运行后,查看csv文件内容:

$ cat /tmp/test1.csv
���,"����
�ʹ�Ӣ�Ķ���,"
����,�й�

使用 WPS 打开文件不会乱码,因为会自动识别文件编码:

参考

https://www.baeldung.com/java-csv


( 本文完 )

文章目录