Fortran 相当于 numpy.where() 函数?

2022-01-14 00:00:00 numpy fortran slice

问题描述

我想在 Fortran 中做这样的事情:

I would like to do something like this in Fortran:

program where

real :: a(6) = (/ 4, 5, 6, 7, 8, 9 /)

print *, a(a>7)

end program

在 Python 我通常会使用 NumPy 像这样:

In Python I would typically do this with NumPy like this:

import numpy

a = numpy.array([ 4, 5, 6, 7, 8, 9])

print a[numpy.where(a>7)]

#or 

print a[a>7]

我玩过,但到目前为止没有任何效果,但我猜它相当简单.

I've played around, but nothing has worked thus far, but I'm guessing it is fairly simple.


解决方案

我将稍微扩展@VladimirF 的答案,因为我怀疑您不想将自己限制在确切的打印示例中.

I'll extend slightly the answer by @VladimirF as I suspect you don't want to limit yourself to the exact print example.

a>7 返回一个与 a 对应的 logical 数组,其中 .true. 位于条件的索引处满足,.false. 否则.pack 内部函数采用这样的掩码并返回一个数组,其中包含掩码中带有 .true. 的那些元素.

a>7 returns a logical array corresponding to a with .true. at index where the condition is met, .false. otherwise. The pack intrinsic takes such a mask and returns an array with those elements with .true. in the mask.

但是,您可以使用可能适合您 numpy.where 愿望的掩码做其他事情.例如,有 where 构造(和 where 语句)和 merge 内在函数.此外,您可以再次使用 pack 和掩码来获取索引并进行更多相关操作.

However, you can do other things with the mask which may fit under your numpy.where desire. For example, there is the where construct (and where statement) and the merge intrinsic. Further you can use pack again with the mask to get the indices and do more involved manipulations.

相关文章