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 中那样进行转换的方法。我可以使用那些线上的第 3 方库或其他东西吗？任何方向都会对我有很大帮助。谢谢大家。

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 文件，并在根据逗号（这是您的分隔符）拆分内容后将内容粘贴到字符串数组中。然后它将它们添加到数组列表中。我知道这可能并不完美，所以之后我获取了该数组列表的内容并将其转换为一个整洁的 2D 矩阵。希望这可以帮助。

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