Skip to content

Commit 26a4c03

Browse files
committed
fixed sort_rows func
1 parent 616bd9f commit 26a4c03

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

CCA.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -935,9 +935,13 @@ def save_results(X: np.ndarray, Y: np.ndarray, Z: np.ndarray, R: np.ndarray, ite
935935
writer.writerow([X[i], Y[i], Z[i], R[i]])
936936

937937
def sort_rows(i_orden: np.ndarray):
938-
for i in range(i_orden.shape[1]):
939-
temp = i_orden[0,i]
940-
i_orden[0,i] = i_orden[-1,i]
941-
i_orden[-1,i] = temp
938+
c_sort = 2
939+
for irow in range(i_orden.shape[0]):
940+
krow = np.argmin(i_orden[irow:i_orden.shape[0],c_sort]) + irow
942941

942+
temp = copy.deepcopy(i_orden[irow,:])
943+
i_orden[irow,:] = i_orden[krow,:]
944+
i_orden[krow,:] = temp
945+
946+
print(f"sortrows = {i_orden}")
943947
return i_orden

0 commit comments

Comments
 (0)