Java中的CSV生成

城南|阿洋-计算机从小白到大神

于 2024-07-17 13:05:59 发布

阅读量886

点赞数 17

文章标签： java python 开发语言

本文链接：https://blog.csdn.net/weixin_46372265/article/details/140492048

版权

爆款标题

揭秘：如何在Java中快速生成高效的CSV文件

大家好,我是城南。今天我们来聊聊如何在Java中生成CSV文件。CSV文件作为一种广泛使用的数据交换格式，在各种系统和应用程序之间扮演着重要角色。无论是导出数据报表，还是进行数据迁移，掌握CSV文件的生成方法都是一项必备技能。下面，我们将深入探讨在Java中生成CSV文件的各种方法，并附带一些常见库的使用技巧。

前言

在如今这个数据为王的时代，如何高效地生成和处理CSV文件成了程序员们必备的技能之一。对于Java开发者而言，生成CSV文件的需求更是时常遇见。话不多说，让我们直接进入主题，探讨几种在Java中生成CSV文件的方式。

方法一：使用Java内置的IO类

首先，我们可以使用Java内置的java.io包来创建和写入CSV文件。这种方法相对基础，但却十分实用。

示例代码

import java.io.BufferedWriter;
import java.io.FileWriter;
import java.io.IOException;

public class CsvWriterSimple {
    public static void main(String[] args) {
        String csvFile = "output.csv";
        try (BufferedWriter writer = new BufferedWriter(new FileWriter(csvFile))) {
            String header = "Name, Age, Gender";
            String record1 = "Alice, 30, Female";
            String record2 = "Bob, 25, Male";
            writer.write(header);
            writer.newLine();
            writer.write(record1);
            writer.newLine();
            writer.write(record2);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

在这个例子中，我们使用BufferedWriter和FileWriter来写入CSV文件。write方法逐行写入数据，最后通过newLine方法换行。

方法二：使用OpenCSV库

OpenCSV是一个流行的CSV处理库，它提供了简洁易用的API来读取和写入CSV文件。使用OpenCSV可以大大简化代码量，同时提高可读性。

导入依赖

在使用OpenCSV之前，我们需要在项目中引入依赖。对于Maven项目，可以在pom.xml中添加以下依赖：

<dependency>
    <groupId>com.opencsv</groupId>
    <artifactId>opencsv</artifactId>
    <version>5.5.2</version>
</dependency>

示例代码

import com.opencsv.CSVWriter;
import java.io.FileWriter;
import java.io.IOException;

public class OpenCsvWriter {
    public static void main(String[] args) {
        String csvFile = "output.csv";
        try (CSVWriter writer = new CSVWriter(new FileWriter(csvFile))) {
            String[] header = {"Name", "Age", "Gender"};
            String[] record1 = {"Alice", "30", "Female"};
            String[] record2 = {"Bob", "25", "Male"};
            writer.writeNext(header);
            writer.writeNext(record1);
            writer.writeNext(record2);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

使用OpenCSV，我们只需调用writeNext方法即可将数组数据写入CSV文件，简洁又高效。

方法三：使用Apache Commons CSV库

Apache Commons CSV是另一个强大的CSV处理库，支持多种CSV格式的读写操作，并提供了丰富的功能来处理复杂的CSV文件需求。

导入依赖

同样地，我们需要在项目中引入Apache Commons CSV的依赖：

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

示例代码

import org.apache.commons.csv.CSVFormat;
import org.apache.commons.csv.CSVPrinter;
import java.io.FileWriter;
import java.io.IOException;
import java.util.Arrays;

public class ApacheCsvWriter {
    public static void main(String[] args) {
        String csvFile = "output.csv";
        try (CSVPrinter printer = new CSVPrinter(new FileWriter(csvFile), CSVFormat.DEFAULT)) {
            printer.printRecord("Name", "Age", "Gender");
            printer.printRecord("Alice", 30, "Female");
            printer.printRecord("Bob", 25, "Male");
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

在这个例子中，Apache Commons CSV通过CSVPrinter类提供了简洁的API，使我们能够以更灵活的方式写入CSV文件。

方法四：自定义CSV格式

有时候，我们需要生成自定义格式的CSV文件，例如需要特定的分隔符或定制的字段处理逻辑。下面展示一种通过Java流式API实现自定义CSV格式的方法。

示例代码

import java.io.BufferedWriter;
import java.io.FileWriter;
import java.io.IOException;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;

public class CustomCsvWriter {
    private static final String DELIMITER = ";";

    public static void main(String[] args) {
        List<String[]> data = Arrays.asList(
                new String[]{"Name", "Age", "Gender"},
                new String[]{"Alice", "30", "Female"},
                new String[]{"Bob", "25", "Male"}
        );

        String csvFile = "output.csv";
        try (BufferedWriter writer = new BufferedWriter(new FileWriter(csvFile))) {
            for (String[] record : data) {
                String line = Arrays.stream(record)
                        .map(CustomCsvWriter::escapeSpecialCharacters)
                        .collect(Collectors.joining(DELIMITER));
                writer.write(line);
                writer.newLine();
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    private static String escapeSpecialCharacters(String data) {
        String escapedData = data.replaceAll("\\R", " ");
        if (data.contains(DELIMITER) || data.contains("\"") || data.contains("'")) {
            data = data.replace("\"", "\"\"");
            escapedData = "\"" + data + "\"";
        }
        return escapedData;
    }
}