使用numpy和swig将2d数组作为参数传递给函数

pip install numpy
pip install swig

import numpy as np

def my_function(arr):
    # 在这里进行对数组的处理
    # 例如，打印数组的形状
    print("Array shape:", arr.shape)

# 创建一个2D数组
arr = np.array([[1, 2, 3], [4, 5, 6]])

# 调用函数，并传递2D数组作为参数
my_function(arr)

%module my_function

%{
#include "my_function.h"  // 包含C/C++函数的头文件
%}

%include "numpy.i"

%init %{
import_array();  // 初始化numpy模块
%}

%apply (int* IN_ARRAY2, int DIM1, int DIM2) {(int* arr, int size1, int size2)};  // 应用参数的转换规则

%include "my_function.h"  // 包含C/C++函数的头文件

#ifndef MY_FUNCTION_H
#define MY_FUNCTION_H

void my_function(int* arr, int size1, int size2);

#endif

#include <iostream>

void my_function(int* arr, int size1, int size2) {
    // 在这里对传入的数组进行处理
    // 例如，打印数组的值
    for (int i = 0; i < size1; i++) {
        for (int j = 0; j < size2; j++) {
            std::cout << arr[i * size2 + j] << " ";
        }
        std::cout << std::endl;
    }
}

swig -python -c++ my_function.i
g++ -c -fpic my_function_wrap.cxx -I /usr/include/python2.7  # 根据Python版本和系统环境修改路径
g++ -shared my_function_wrap.o -o _my_function.so

import my_function

# 创建一个2D数组
arr = np.array([[1, 2, 3], [4, 5, 6]])

# 调用函数，并传递2D数组作为参数
my_function.my_function(arr)