数据排序

Kieran · 发表于 2023-6-16 09:55:32

本帖最后由 Kieran 于 2023-6-19 10:15 编辑

大家好，

我有一个数据文件。内容如下。

[Fortran] 纯文本查看 复制代码

01         116
02Parameter
03   0.2951429939E+02    0.0000000000E+00    0.0000000000E+00
04   0.0000000000E+00    0.1500000000E+02    0.0000000000E+00
05   0.0000000000E+00   0.0000000000E+00   0.2510000038E+02
06Parameter
07   0.000000000    0.500000000    0.300000012
08   0.041669998    0.500000000    0.328289986
09   0.041669998    0.500000000    0.384860009
10   0.000000000    0.500000000    0.413150012
11   0.000000000    0.500000000    0.469720006
12   0.041669998    0.500000000    0.498010010
13   0.041669998    0.500000000    0.554579973
14   0.000000000    0.500000000    0.582870007
15   0.083329998    0.500000000    0.300000012
16   0.125000000    0.500000000    0.328289986
17   0.125000000    0.500000000    0.384860009
18   0.083329998    0.500000000    0.413150012
19   0.083329998    0.500000000    0.469720006
20   0.125000000    0.500000000    0.498010010
21。。。。。。

我想重排各行的位置。重排标准是，先按每行中的第一数据，由小到达排列。如果有两行，或者多行的第一个数据相等，则按第二个数据由小到大重排这些行的位置。如果有两行，或者多行第二个数据也相等，则按第三个数据重排各行位置。

这是我的代码。

[Fortran] 纯文本查看 复制代码

01PROGRAM COMPAIMPLICIT NONE
02 
03INTEGER, PARAMETER :: dp = SELECTED_REAL_KIND(15,14)
04INTEGER            :: i, j, k, l
05REAL (KIND=dp)     :: ap(116,6), lr(3,3), pl(3), ns(2,116)
06 
07OPEN (UNIT=3, FILE='data.dat', STATUS='OLD')
08READ (UNIT=3, FMT=*)
09READ (UNIT=3, FMT=*)
10DO i = 1, 3, 1
11   READ (UNIT=3, FMT=*) pl
12   lr(i,:) = pl
13END DO
14READ (UNIT=3, FMT=*)
15DO i = 1, 116, 1
16   READ (UNIT=3, FMT=*) pl
17   ap(i,4:6) = pl
18   ns(:,i) = i
19END DO
20 
21DO i = 1, 116, 1
22   ap(i,1) = ap(i,4)*lr(1,1)+ap(i,5)*lr(2,1)+ap(i,6)*lr(3,1)
23   ap(i,2) = ap(i,4)*lr(1,2)+ap(i,5)*lr(2,2)+ap(i,6)*lr(3,2)
24   ap(i,3) = ap(i,4)*lr(1,3)+ap(i,5)*lr(2,3)+ap(i,6)*lr(3,3)
25END DO
26 
27DO i = 2, 116, 1
28   DO j = 1, i-1, 1
29      IF ((ap(i,1) < ap(j,1)) .OR. &
30         ((ap(i,1) == ap(j,1)) .AND. (ap(i,2) < ap(j,2))) .OR. &
31         ((ap(i,1) < ap(j,1)) .AND. (ap(i,2) == ap(j,2)) .AND. &
32         (ap(i,3) < ap(j,3)))) THEN
33         k = ns(1,i)
34         DO l = j, i-1, 1
35            ns(1,l+1) = ns(1,l)
36         END DO
37         ns(1,j) = k
38      END IF
39   END DO
40END DO
41 
42OPEN (UNIT=4, FILE='compare.dat', STATUS='UNKNOWN')
43DO i = 1, 116, 1
44   DO j = 1, 116, 1
45      IF (ns(1,j) == i) WRITE (UNIT=4, FMT=*) ns(:,j), ap(j,1:3)
46   END DO
47END DO
48 
49CLOSE (UNIT=3)
50CLOSE (UNIT=4)
51STOP
52END PROGRAM COMPA

运行后，输出的重排结果如下。

[Fortran] 纯文本查看 复制代码

01    1.0000000000000000        1.0000000000000000        0.0000000000000000        7.5000000000000000        7.5300004152000044        91.000000000000000        3.0000000000000000        1.2298607965527013        7.5000000000000000        9.6599863721468040    
02   91.000000000000000        4.0000000000000000        0.0000000000000000        7.5000000000000000        10.370065458197006    
03   91.000000000000000        5.0000000000000000        0.0000000000000000        7.5000000000000000        11.789972329093603    
04   91.000000000000000        6.0000000000000000        1.2298607965527013        7.5000000000000000        12.500051440243805    
05   91.000000000000000        7.0000000000000000        1.2298607965527013        7.5000000000000000        13.919957533040389    
06   91.000000000000000        8.0000000000000000        0.0000000000000000        7.5000000000000000        14.630037397190604    
07   91.000000000000000        9.0000000000000000        2.4594265091401013        7.5000000000000000        7.5300004152000044    
08   92.000000000000000        2.0000000000000000        1.2298607965527013        7.5000000000000000        8.2400787733501950    
09   93.000000000000000        11.000000000000000        3.6892874237500002        7.5000000000000000        9.6599863721468040
10。。。。。。

很显然，这个排序时错误的，但我没找到程序里的错误。能麻烦大家帮我看下，我的代码哪里出了错误码？谢谢。

谢谢。

necrohan · 发表于 2023-6-20 10:02:11

你的意思是不是用数组ns保存排序的序号，我大概修改了一下

[Fortran] 纯文本查看 复制代码

01! nline=116
02DO i = 1, nline-1
03  write(*,'(2i4)')i;
04  ins=ns(1,i); ! 数组排序序号，原始序号在ns(2,i)中
05   DO j = i+1, nline
06      jns=ns(1,j); ! 序号
07      IF ((ap(jns,1) < ap(ins,1)) .OR. &
08         ((abs(ap(jns,1)-ap(ins,1))<1.0e-6) .AND. (ap(jns,2) < ap(ins,2))) .OR. &
09         ((abs(ap(jns,1)-ap(ins,1))<1.0e-6) .AND. (abs(ap(jns,2)-ap(ins,2))<1.0e-6) .AND. (ap(jns,3) < ap(ins,3))) & ! TODO: 原来是<
10         ) THEN
11         write(*,*)'swap ',i,j
12         ns(1,i) = jns
13         ns(1,j)=ins
14         ins=jns
15         cycle;
16      END IF
17   END DO
18END DO
19 
20OPEN (UNIT=4, FILE='resu.txt')
21DO i = 1, nline
22   WRITE (UNIT=4, FMT='(2i6,2x,10(f12.4,2x))') ns(1:2,i), ap(ns(1,i),1:3)
23END DO

Kieran · 发表于 2023-6-20 11:55:15

本帖最后由 Kieran 于 2023-6-20 12:06 编辑

necrohan 发表于 2023-6-20 10:02
你的意思是不是用数组ns保存排序的序号，我大概修改了一下
[mw_shl_code=fortran,true]
! nline=116

非常感谢你的回复。

我的代码确实是在用ns变量保存各行在排序后的新序号。

我想请教几个问题。

1.你的代码的比较方式，是将各行中”最小“的那一行放在序号变量ns中的最前面，然后再放第二小的，紧接着是第三小，第四小。。。以此类推。我的理解正确吗？

2.请问你的代码中的CYCLE命令作用是什么呢？可不可以不用这个命令呢？

3.请问你觉得我的代码中哪里出了错误呢？我一直没找到原因。

谢谢，望指教。

necrohan · 发表于 2023-6-20 15:06:12

Kieran 发表于 2023-6-20 11:55
非常感谢你的回复。

我的代码确实是在用ns变量保存各行在排序后的新序号。

1、我能大概明白你的代码的意思，但是排序那里和我思路不一样，我参考你的方式按自己的思路写的，结果就是你说的那样，从小到大排列。
2、cycle是直接进行下一次循环，不执行endif后面的语句节省点时间，这里因为endif后面没有语句了，所以不用也可以。
3、如果读入的lr是单位矩阵，那么你的代码ap(i,1:3)和ap(i,4:6)是相同的，我是这么测试查看输出结果的。
你的代码我没太明白，但是31行那个ap(i,1)<ap(j,1)应该是==。
我印象中浮点数比较用==的容易出问题，我用的绝对值差判断。
你的代码34～36行我不明白为什么要对ns循环赋值，我理解ns数组保存的序号，直接调换就行。
输出排序结果那段循环，就是43～47行，我不明白一共就116行你为什么要嵌套2次，直接循环116次输出就行。

Kieran · 发表于 2023-6-20 15:16:59

necrohan 发表于 2023-6-20 15:06
1、我能大概明白你的代码的意思，但是排序那里和我思路不一样，我参考你的方式按自己的思路写的，结果就 ...

谢谢你的回复和指点。

我原本想着用insertion sorting方法排序，但我可能自己没想清楚每个步骤，所以代码不合理。

我再仔细想想吧。

再次感谢。

[通用算法] 数据排序

规矩勋章