csv 处理 java

原创

mob64ca12e3a791 2024-08-26 06:06:01 ©著作权

©著作权归作者所有：来自51CTO博客作者mob64ca12e3a791的原创作品，请联系作者获取转载授权，否则将追究法律责任

学习使用 Java 处理 CSV 文件

处理 CSV（逗号分隔值）文件是很多应用程序中常见的任务。在这篇文章中，我们将深入探讨如何使用 Java 处理 CSV 文件。我们将逐步理解整个流程，并提供相应的代码示例和解释。

流程概述

在处理 CSV 文件的过程中，我们需要遵循以下步骤：

步骤	描述
1	准备工作：设置开发环境和依赖
2	读取 CSV 文件
3	处理数据
4	写入新 CSV 文件

流程图

以下是整个流程的图示：

flowchart TD
    A[准备工作] --> B[读取 CSV 文件]
    B --> C[处理数据]
    C --> D[写入新 CSV 文件]

每一步的详细说明

接下来，我们将逐步详解每一个步骤。

步骤 1: 准备工作

确保你有一个 Java 开发环境，比如安装 IntelliJ IDEA 或 Eclipse，并且确保 Java SDK 已经安装。

如果你需要使用 CSV 库，建议使用 OpenCSV，所以在项目中添加该依赖（如果使用 Maven，添加如下依赖）：

<dependency>
    <groupId>com.opencsv</groupId>
    <artifactId>opencsv</artifactId>
    <version>5.5.2</version> <!-- 或者使用最新版本 -->
</dependency>

步骤 2: 读取 CSV 文件

我们将创建一个方法来读取 CSV 文件，并将其内容存储在列表中。

import com.opencsv.CSVReader;
import java.io.FileReader;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

// 读取 CSV 文件的方法
public List<String[]> readCSV(String filePath) {
    List<String[]> data = new ArrayList<>();
    
    try (CSVReader reader = new CSVReader(new FileReader(filePath))) {
        String[] line;
        while ((line = reader.readNext()) != null) {
            data.add(line); // 将每行数据存入列表
        }
    } catch (IOException e) {
        e.printStackTrace(); // 打印异常信息
    }
    
    return data; // 返回读取到的数据
}

步骤 3: 处理数据

接下来，我们需要处理读取到的数据。这里我们可以根据需求进行过滤、转换或其它操作。以下是一个示例方法，用于将数据转换为大写字母并返回。

// 处理数据的方法
public List<String[]> processData(List<String[]> rawData) {
    List<String[]> processedData = new ArrayList<>();
    
    for (String[] line : rawData) {
        String[] processedLine = new String[line.length];
        for (int i = 0; i < line.length; i++) {
            processedLine[i] = line[i].toUpperCase(); // 将每个单元格内容转为大写
        }
        processedData.add(processedLine); // 加入处理后的数据
    }
    
    return processedData; // 返回处理后的数据
}

步骤 4: 写入新 CSV 文件

最后，我们将处理后的数据写入到一个新的 CSV 文件中。

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

// 写入 CSV 文件的方法
public void writeCSV(String filePath, List<String[]> data) {
    try (CSVWriter writer = new CSVWriter(new FileWriter(filePath))) {
        writer.writeAll(data); // 将所有数据写入文件
    } catch (IOException e) {
        e.printStackTrace(); // 打印异常信息
    }
}

整合步骤

我们现在可以将所有的方法整合到一个完整的 Java 类中：

import com.opencsv.CSVReader;
import com.opencsv.CSVWriter;
import java.io.FileReader;
import java.io.FileWriter;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

public class CsvProcessor {

    public List<String[]> readCSV(String filePath) {
        // ... 上面第一部分的代码 ...
    }

    public List<String[]> processData(List<String[]> rawData) {
        // ... 上面第二部分的代码 ...
    }

    public void writeCSV(String filePath, List<String[]> data) {
        // ... 上面第三部分的代码 ...
    }

    public static void main(String[] args) {
        CsvProcessor processor = new CsvProcessor();
        
        List<String[]> rawData = processor.readCSV("input.csv"); // 读取输入的 CSV 文件
        List<String[]> processedData = processor.processData(rawData); // 处理数据
        processor.writeCSV("output.csv", processedData); // 写入结果到新的 CSV 文件
    }
}

序列图示

以下是整个流程的序列图：

sequenceDiagram
    participant A as 用户
    participant B as CsvProcessor
    A->>B: 调用 readCSV("input.csv")
    B-->>A: 返回原始数据
    A->>B: 调用 processData(rawData)
    B-->>A: 返回处理后的数据
    A->>B: 调用 writeCSV("output.csv", processedData)
    B-->>A: 数据写入完成