Java将csv文件读取为矩阵

Question

我是编写 Java 代码的新手。我有使用脚本类型语言编写代码的经验。我正在尝试用 java 重写我在 python 中的一段代码。

下面的 Python 代码 -

import pandas as pd
myFile = 'dataFile'
df = pd.DataFrame(pd.read_csv(myFile,skiprows=0))
inData = df.as_matrix()

我正在java中寻找一种与python中的as_matrix等效的方法。该函数将数据帧转换为矩阵。

我确实查找了一段时间，但找不到像 python 中那样进行转换的方法。是否有第三方库或我可以使用的相关内容？任何方向都会对我有很大帮助。谢谢大家。

Answer 1

您想要做的事情非常简单，并且需要最少的代码，因此我建议您自己编写代码。这是一个示例实现:

List<String[]> rowList = new ArrayList<String[]>();
try (BufferedReader br = new BufferedReader(new FileReader("pathtocsvfile.csv"))) {
    String line;
    while ((line = br.readLine()) != null) {
        String[] lineItems = line.split(",");
        rowList.add(lineItems);
    }
    br.close();
}
catch(Exception e){
    // Handle any I/O problems
}
String[][] matrix = new String[rowList.size()][];
for (int i = 0; i < rowList.size(); i++) {
    String[] row = rowList.get(i);
    matrix[i] = row;
}

它的作用非常简单:它打开一个缓冲读取器，该读取器将逐行读取 csv 文件，并在根据逗号(这是分隔符)分割内容后将内容粘贴到字符串数组中。然后它将它们添加到数组列表中。我知道这可能并不完美，所以后来我获取该数组列表的内容并将其转换为一个整洁的二维矩阵。希望这可以帮助。

提示:可以对这一小段代码进行很多改进(即处理尾随和前导空格、添加用户定义的分隔符等)，但这应该是一个很好的起点。