我们之前讨论过如何用python在三维矩阵中切片，在Python中保存三维矩阵和它的三个坐标轴的方法有用numpy.savez将数组和矩阵用不同的键值对保存至一个.npz文件。这里将讨论如何用Fortran实现这一过程。

NPY-for-Fortran：`npy`文件的Fortran接口#

NPY-for-Fortran为Fortran提供了npy和npz的接口模块。我们在Fortran中将多种精度的intger, real和cmplx的一维或多维数组保存至npy或npz文件。

先将仓库克隆至本地

1
git clone https://github.com/xumi1993/NPY-for-Fortran.git

然后复制src/npy.f90到任意可调用的路径。

调用NPY-for-Fortran#

Fortran程序实例#

这里我们定义一个三维矩阵mtx，和它的三个坐标轴x、y和z，然后将他们保存在一个npz文件里

1
program main
2
    use m_npy
3

4
    integer     :: i,j,k
5
    real(8)     :: mtx(9,10,11), z(9), y(10), x(11)
6
    character(len=10) :: fname
7

8
    fname = 'test.npz'
9
    do i=1,9
10
        z(i) = i*0.3
11
        do j=1,10
12
            y(i) = i*0.4
13
            do k=1,11
14
                x(j) = j *0.5
15
                mtx(i,j,k) = 20*i**2 + 100*j + k
16
            enddo
17
        enddo
18
    enddo
19

20

21
    call add_npz(fname, 'x', x)
22
    call add_npz(fname, 'y', y)
23
    call add_npz(fname, 'z', z)
24
    call add_npz(fname, 'mtx', mtx)
25

26
end program

编译#

由于npy.f90是一个Fortran模块，所以要先编译这个模块，生成.mod文件，再编译上面这个主程序

1
gfortran npy.f90 -c
2
gfortran example.f90 npy.f90 -o example_npz

其中example.f90是上述主程序。

运行#

运行程序后会生成一个test.npz文件。

1
./example_npz

在Python中查看#

用numpy.load即可打开npz文件并在Python脚本中调用其数值。

1
import numpy as np
2
data = np.load('test.npz')
3
print(data.__dict__)

结果为

1
{'_files': ['x.npy', 'y.npy', 'z.npy', 'mtx.npy'],
2
 'files': ['x', 'y', 'z', 'mtx'],
3
 'allow_pickle': False,
4
 'pickle_kwargs': {'encoding': 'ASCII', 'fix_imports': True},
5
 'zip': <zipfile.ZipFile file=<_io.BufferedReader name='test.npz'> mode='r'>,
6
 'f': <numpy.lib.npyio.BagObj at 0x7fc3b819c4c0>,
7
 'fid': <_io.BufferedReader name='test.npz'>}

NPY-for-Fortran：npy文件的Fortran接口#

调用NPY-for-Fortran#

Fortran程序实例#

编译#

运行#

在Python中查看#

NPY-for-Fortran：`npy`文件的Fortran接口#