python - 如何使用 ctypes 将 numpy ndarrays 传递给 c++ 函数?
问题描述
我花了两天时间弄清楚如何将 ndarrays 传递给 c/c++,但仍然没有找到任何可以正常工作的东西。
每件事都被记录得很糟糕而且令人困惑。
我发现的唯一教程在https://scipy.github.io/old-wiki/pages/C%2B%2B_Extensions_that_use_NumPy_arrays.html中。
但是当我尝试运行 python 代码“myextension.py”时,它会引发 ImportError:没有名为 numpyndarrays 的模块。
我不想使用 cython 或 boost 或 python C-API。
这是我编译到共享库的 c++ 代码“n.cpp”:
#include <iostream>
#include "ndarray.h"
extern "C" {
Ndarray<double, 3> myfunc(numpyArray<double> array1)
{
Ndarray<double,3> a(array1);
std::cout<<a.getShape(0); //this prints 0 and it means somthing is wrong
for (int i = 0; i < a.getShape(0); i++)
{
for (int j = 0; j < a.getShape(1); j++)
{
for (int k = 0; k < a.getShape(2); k++)
{
a[i][j][k] = 2.0 * a[i][j][k];
std::cout<<a[i][j][k]<<' '<<i<<' '<<j<<' '<<k<<'\n';// prints nothing because a.getShape(0) is 0.
}
}
}
return a;
}
}
起初,当我试图编译这段代码时,编译器引发了致命错误,因为它找不到 ndarray.h。我从https://scipy.github.io/old-wiki/pages/ndarray2e5c.h?action=AttachFile&do=get&target=ndarray.h
手动下载了 ndarray.h 文件
这是我的python代码:
import numpy as np
import numpy.ctypeslib as ct
mylib = ct.load_library('n', '.')
def myfunc(array1):
return mylib.myfunc(array1.ctypes) # I also tried ct.as_ctypes(array1) instead of array1.ctypes with no success
a = np.ones((2,2,2)).astype(np.double)
myfunc(a)
此代码正确加载共享库,主要问题是如何将 numpy ndarray 转换为作为共享库输入的结构。
我还需要知道,如何将共享库的输出转换为 numpy ndarray。
解决方案
我发现使用 numpy C-API 是将 numpy 数组传递给 c++ 的最简单方法。
该文档可以在https://docs.scipy.org/doc/numpy/reference/c-api.html上找到。
推荐阅读
- c# - Unity Photon 定时器同步
- database-design - 如何使用 cassandra 实现更改流?
- azure-devops - Devops terraform:.tf 文件中的后端配置与 devops 的 terraform CD 配置中的后端配置
- sql - 将同一表中的查询合并到不同的列中
- linux - Linux命令在文件中的某个单词之后剪切所有内容并导出到另一个文件
- arrays - 为什么我会从二维数组中打印出随机数?
- javascript - 如何比较一天中某个时间过去的时间(超过 23:59)
- python - 嵌套 if 在不工作的 Python 中有一个 while 循环
- prometheus - 聚合运算符中使用的每个标签值的即时向量
- sql - 通过传递动态表名和列名来计算重复记录