HOWTO · Java
用 Java 将字符串写入 CSV 文件
本文介绍了如何使用两种方法在 Java 中将字符串写入 CSV。
本页内容
CSV 代表 Comma Separated Values；它是系统之间大量数据传输的常用格式。还有许多与 Java 一起使用的解析器库能够处理这种格式。
在 Java 中使用 PrintWriter 读取和写入 CSV 文件
Java 中的 PrinterWriter 函数是一个编写器类，用于将对象的格式化表示打印到文本输出流。我们创建了一个 writer 对象，传递一个名为 test.csv 的新文件作为 writer 的目的地。在这里，sb 对象将指定的字符串附加到字符序列。
writer 实例上的 write() 方法将文本内容写入流中。flush() 方法将内容刷新到文件中，close() 方法永久关闭流。我们还可以读取 test.csv 文件的内容。
在 TestCSV 类实例上调用 readCSVFile() 方法。在这里，我们使用指定文件中的值创建了 Scanner 类的实例。它使用分隔符模式将数据分解为标记。如果 Scanner 有另一行输入，则 hasNextLine() 返回 true。因此，我们使用 next() 方法读取每一行标记数据并将其存储在一个数组中，最后打印输出。
import java.io.File;
import java.io.FileNotFoundException;
import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;
public class TestCSV {
public static void main(String args[]) {
try (PrintWriter writer = new PrintWriter(new File("test.csv"))) {
StringBuilder sb = new StringBuilder();
sb.append("id");
sb.append(',');
sb.append("Name");
sb.append(',');
sb.append("Address");
sb.append('\n');
sb.append("101");
sb.append(',');
sb.append("John Doe");
sb.append(',');
sb.append("Las Vegas");
sb.append('\n');
writer.write(sb.toString());
writer.close();
System.out.println("done!");
} catch (FileNotFoundException e) {
System.out.println(e.getMessage());
}
TestCSV testCSV = new TestCSV();
testCSV.readCSVFile();
}
public void readCSVFile() {
List<List<String>> records = new ArrayList<>();
try (Scanner scanner = new Scanner(new File("test.csv"));) {
while (scanner.hasNextLine()) {
records.add(getRecordFromLine(scanner.nextLine()));
}
} catch (FileNotFoundException e) {
e.printStackTrace();
}
System.out.println(records.toString());
}
private List<String> getRecordFromLine(String line) {
List<String> values = new ArrayList<String>();
try (Scanner rowScanner = new Scanner(line)) {
rowScanner.useDelimiter(",");
while (rowScanner.hasNext()) {
values.add(rowScanner.next());
}
}
return values;
}
}
输出：
done!
[[id, Name, Address], [101, John Doe, Las Vegas]]
在 Java 中使用 OpenCSV 库读取和写入 CSV 文件
这个库的 maven 依赖如下：
<dependency>
<groupId>com.opencsv</groupId>
<artifactId>opencsv</artifactId>
<version>5.4</version>
</dependency>
OpenCsv 是一个简单的 java 解析器库；它有一组 OpenCsv 类，我们用来读取和写入 CSV 文件。在 main() 方法中，我们首先使用 CSVWriter 类调用该方法以写入 CSV 文件。此类用于将 CSV 数据写入编写器实现。
我们创建一个 CSVWriter 的 writer 实例，并在该对象上调用 writeNext() 函数以生成一个 CSV 文件，其中包含使用分隔符分隔的字符串数组中的数据。close() 方法关闭编写器流。
为了从我们以 CSV 格式创建的文件中读取数据，我们在主类对象上调用 readCSVFile 方法，在那里我们解析 CSV 文件。为了一次将所有记录读入列表，我们使用 readAll() 方法。我们遍历每条记录并将其打印出来。
import com.opencsv.CSVReader;
import com.opencsv.CSVWriter;
import com.opencsv.exceptions.CsvException;
import java.io.FileNotFoundException;
import java.io.FileReader;
import java.io.FileWriter;
import java.io.IOException;
import java.util.Arrays;
import java.util.List;
public class CSVLibrary {
public static void main(String args[]) {
CSVLibrary csvLibrary = new CSVLibrary();
csvLibrary.writeToCsv();
csvLibrary.readCSVFile();
}
public void writeToCsv() {
String csv = "data.csv";
try {
CSVWriter writer = new CSVWriter(new FileWriter(csv));
String[] record = "2,Virat,Kohli,India,30".split(",");
writer.writeNext(record);
writer.close();
} catch (IOException e) {
e.printStackTrace();
}
}
public void readCSVFile() {
CSVReader reader = null;
try {
reader = new CSVReader(new FileReader("data.csv"));
} catch (FileNotFoundException e) {
e.printStackTrace();
}
List<String[]> allRows = null;
try {
allRows = reader.readAll();
} catch (IOException e) {
e.printStackTrace();
} catch (CsvException e) {
e.printStackTrace();
}
for (String[] row : allRows) {
System.out.println(Arrays.toString(row));
}
}
}
输出：
[2, Virat, Kohli, India, 30]
