fix the reversed columns and rows #508

LinguaPhylo · Aug 8, 2024 · 6f74889 · 6f74889
1 parent 6290490
commit 6f74889
Showing 1 changed file with 11 additions and 6 deletions.
diff --git a/lphy/src/main/java/lphy/core/model/datatype/Table.java b/lphy/src/main/java/lphy/core/model/datatype/Table.java
@@ -37,14 +37,19 @@ public String[] getColumnNames() {
         return keySet().toArray(new String[0]);
     }
 
-    @MethodInfo(description = "return the 2d matrix of phi and psi angles.")
+    @MethodInfo(description = "return the array of (array) records for selected column indices.")
     public Double[][] getColumnAsMatrix(Integer... ArrayIndex) {
-        Double[][] anglesMatrix = new Double[ArrayIndex.length][];
+
+        List[] columns = new List[ArrayIndex.length];
+        for (int i = 0; i < ArrayIndex.length; i++) {
+            columns[i] = getColumn(ArrayIndex[i]);
+        }
+
+        Double[][] anglesMatrix = new Double[columns[0].size()][ArrayIndex.length];
+
         for (int i = 0; i < anglesMatrix.length; i++) {
-            List column = getColumn(ArrayIndex[i]); //get column 1 and column 2
-            anglesMatrix[i] = new Double[column.size()];
-            for (int j = 0; j < column.size(); j++) {
-                anglesMatrix[i][j] = Double.parseDouble(column.get(j).toString());
+            for (int j = 0; j < anglesMatrix[i].length; j++) {
+                anglesMatrix[i][j] = Double.parseDouble(columns[j].get(i).toString());
             }
         }
         return anglesMatrix;