Fortran中以下Python列表理解的等效项是什么？

Question

我正在尝试在Fortran中编写以下列表理解（用Python编写）。

lit = [[x,y] for x in [p,q,r] for y in [h,k,l] if [x,y]!=[a,b]]

其中a, b, p ,q ,r, h, k, l是整数

如果要先在2D Fortran数组中填充列，如何实现？

Python代码返回一个列表。 相当于

for x in [p,q,r]:
     for y in [h,k,l]:
         if [x,y]!=[a,b]:
             list.append([x,y])

我在Fortran中创建了两个子列表。 sublist_x和sublist_y ，其中每个列表分别包含p,q,r和h,k,l 。

integer :: list(0:7), sublist_x(0:2),sublist_y(0:2), count
 count =-1

do i=0,7
   if (i%3 ==0)
   count = count +1
   endif
   list(0,i)=sublist_x(i%3)
   list(1,i)=sublist_y(count%3)
enddo

我认为这是一种复杂的处理方式...

Answer 1

如果我理解正确，则需要两个小列表的笛卡尔积，但不包括元素[a,b] ？ 如果我误会了，请立即停止阅读。 这是一个几乎可以满足您需要的小程序...

PROGRAM test

  IMPLICIT NONE

  INTEGER, DIMENSION(:), ALLOCATABLE :: listx, listy, bad_element
  INTEGER, DIMENSION(:,:), ALLOCATABLE :: outlist
  INTEGER :: ix, jx, alstat, n_elements

  LOGICAL, DIMENSION(:), ALLOCATABLE :: rows

  listx = [1,2,3]
  listy = [21,22,23]
  bad_element = [3,21]

  n_elements = SIZE(listx)*SIZE(listy)
  ALLOCATE(outlist(2,n_elements),stat=alstat)
  IF (alstat/=0) THEN
     WRITE(*,*) "something went wrong allocating the result array"
     STOP
  ELSE
     outlist(1,:) = RESHAPE(listx,[n_elements],listx)
     outlist(2,:) = RESHAPE(SPREAD(listy,1,SIZE(listx)),[n_elements])
  END IF

  DO ix = 1, n_elements
     IF (ALL(outlist(:,ix)==bad_element)) THEN
        outlist(:,ix:) = EOSHIFT(outlist(:,ix:),1,dim=2)
     END IF
  END DO

END PROGRAM TEST

在该程序的末尾， outlist包含笛卡尔积，其等于坏元素的任何元素都替换为0 s，并推送到outlets的末尾。 对于上面的固定数字，输出为：

    1    2    1    2    3    1    2    3    0
   21   21   22   22   22   23   23   23    0

我想您应该不费吹灰之力地将其删除以消除0 s，也不必将此程序打包到例程中。 我希望代码能够自我解释。