使用 SWIG 的几个 numpy 数组

Question

我正在使用 SWIG 将 numpy 数组从 Python 传递到 C++ 代码：

%include "numpy.i"
%init %{
import_array();
%}

%apply (float* INPLACE_ARRAY1, int DIM1) {(float* data, int n)};

class Class 
{
  public: 
  void test(float* data, int n)
  {
    //...
  }
};

在 Python 中：

c = Class()
a = zeros(5)
c.test(a)

这可行，但如何将多个 numpy 数组传递给同一个函数？

Answer 1

我从同事那里找到了答案：

%apply (float* INPLACE_ARRAY1, int DIM1) {(float* data1, int n1), (float* data2, int n2)};

class Class 
{
  public: 
  void test(float* data1, int n1, float* data2, int n2)
  {
    //...
  }
};

现在两个 numpy 数组被传递给 Class::test。