Ubuntu中Fortran与Python如何交互

在Ubuntu系统中，Fortran和Python可以通过多种方式实现交互。以下是一些常见的方法：

1. 使用f2py

想让Fortran和Python“对话”，f2py是个非常经典的选择。它本质上是NumPy的一个工具，专门负责把Fortran代码“翻译”成Python能直接调用的模块。具体怎么操作？我们一步步来看。

安装f2py

第一步，自然是把环境准备好。你得确保系统里已经装好了NumPy和一个Fortran编译器，比如常用的gfortran。

sudo apt-get update
sudo apt-get install python3-numpy gfortran

编写Fortran代码

接着，创建一个Fortran源文件，比如就叫example.f90。这里的关键是使用iso_c_binding来确保与C语言的兼容性，这是后续交互的基础。

! example.f90
subroutine add(a, b, c) bind(c, name="add")
    use, intrinsic :: iso_c_binding
    real(c_double), intent(in) :: a, b
    real(c_double), intent(out) :: c
    c = a + b
end subroutine add

使用f2py生成Python模块

代码写好了，下一步就是“翻译”。运行下面这条命令，f2py就会开始工作。

f2py -c example.f90 -m example

命令执行成功后，你会得到两个关键文件：一个名为example.so的共享库，以及一个同名的Python接口文件example.py。这意味着，你的Fortran函数已经成功“包装”好了。

在Python中使用生成的模块

现在，在Python脚本里调用它，就跟调用普通Python模块一样简单直观。

import example

a = 1.0
b = 2.0
c = example.add(a, b)
print(f"The result is {c}")

看，整个过程是不是相当顺畅？Fortran的计算能力就这样无缝地融入了Python的工作流。

2. 使用ctypes

如果你更喜欢直接一点的方式，或者对底层交互感兴趣，那么ctypes值得考虑。它是Python标准库的一部分，可以直接调用C语言编写的共享库。思路是：先把Fortran代码编译成C兼容的共享库，再用ctypes去调用它。

编译Fortran代码为C兼容的共享库

首先，我们需要一个C头文件来声明接口。创建一个example.h文件：

// example.h
#ifndef EXAMPLE_H
#define EXAMPLE_H

#ifdef __cplusplus
extern "C" {
#endif

void add_(double *a, double *b, double *c);

#ifdef __cplusplus
}
#endif

#endif // EXAMPLE_H

然后，确保你的Fortran代码（example.f90）使用了正确的绑定名称，以匹配C接口：

! example.f90
subroutine add(a, b, c) bind(c, name="add_")
    use, intrinsic :: iso_c_binding
    real(c_double), intent(in) :: a, b
    real(c_double), intent(out) :: c
    c = a + b
end subroutine add

接下来，就是标准的编译流程了：先编译成目标文件，再链接成共享库。

gfortran -c example.f90 -o example.o
gfortran -shared example.o -o libexample.so

使用ctypes在Python中调用共享库

共享库准备就绪，Python这边就可以动手了。使用ctypes需要明确指定函数的参数和返回类型，这一步虽然稍显繁琐，但能让你对数据传递有完全的控制。

import ctypes

# 加载共享库
libexample = ctypes.CDLL('./libexample.so')

# 定义函数的参数和返回类型
libexample.add_.argtypes = [ctypes.c_double, ctypes.c_double, ctypes.POINTER(ctypes.c_double)]
libexample.add_.restype = None

# 调用函数
a = ctypes.c_double(1.0)
b = ctypes.c_double(2.0)
c = ctypes.c_double()
libexample.add_(ctypes.byref(a), ctypes.byref(b), ctypes.byref(c))

print(f"The result is {c.value}")

这种方法虽然步骤多一些，但胜在直接和灵活，尤其适合需要精细控制内存和指针的场景。

3. 使用pyfortran

最后，我们来看一个相对较新的工具——pyfortran。它的目标更“激进”一些：试图将Fortran代码直接转换为等价的Python代码。这对于想要快速迁移或理解逻辑的人来说，可能是一条捷径。

安装pyfortran

安装过程很简单，一条pip命令搞定。

pip install pyfortran

编写Fortran代码

我们继续用那个加法函数的例子，Fortran代码和之前类似：

! example.f90
subroutine add(a, b, c) bind(c, name="add")
    use, intrinsic :: iso_c_binding
    real(c_double), intent(in) :: a, b
    real(c_double), intent(out) :: c
    c = a + b
end subroutine add

使用pyfortran生成Python代码

然后，运行转换命令。注意，这里的输出直接就是Python模块文件。

pyfortran example.f90 -o example.py

在Python中使用生成的代码

使用起来，体验和f2py生成的模块非常相似，都是直接导入调用。

import example

a = 1.0
b = 2.0
c = example.add(a, b)
print(f"The result is {c}")

当然，需要提醒的是，pyfortran对Fortran语法的支持范围可能不如f2py那么全面，对于复杂的项目，建议先进行充分测试。

好了，以上就是几种在Ubuntu系统中让Fortran与Python协同工作的主流方法。从经典的f2py，到底层的ctypes，再到试图直接转换的pyfortran，每种方案都有其适用场景。具体选择哪一个，就得看你项目的具体需求、对性能的控制要求以及对工作流程的偏好了。