python - C++ Boost Python numpy 数组初始化

Question

在 Python 中，我可以用负 inf 初始化 x 长度的 numpy 数组

import numpy as np
...
foo = np.array([np.NINF] * x)

哪里x是一个整数42. 我想在 C++ 中用 Boost.Python 做同样的事情。以下显然行不通:

namespace bnp = boost::python::numpy;
...
bnp::ndarray foo = bnp::array({-INFINITY} * x);

有哪些好的方法可以做到这一点？

是的，我知道 Boost.Numpy docs and tutorial -- 他们不是很好。

更一般地说，我如何用值 -INFINITY 初始化一个标准 vector 或长度为 x 的数组？

更新:

我试图通过打印到控制台来验证方法(使用评论中建议的初始循环)

for (auto i=0; i<x; ++i) {
    std::cout << foo[i] << '\n';
}

但出现以下错误:error: use of overloaded operator '<<' is ambiguous (with operand types 'ostream' (aka 'basic_ostream<char>') and 'object_item' (aka 'proxy<boost::python::api::item_policies>')) .为什么这行不通？尝试按索引访问 boost numpy 数组是否有问题？

Answer 1

这是一个解决方案(感谢@DanMašek 的最初想法)以及如何通过打印到控制台进行验证:

bpy::list temp_list;
temp_list.append(-INFINITY);
temp_list *= x;
bnp::ndarray foo = bnp::array(temp_list);

我有 x=9。验证/

std::cout << std::endl << "Python ndarray : " << bpy::extract<char const *>(bpy::str(foo)) << std::endl;

您还可以使用相同的 temp_list 来初始化另一个 Python ndarray:

// after initializing bar the same as foo w/ temp_list
bar[0] = 0;
std::cout << std::endl << "Python ndarray : " << bpy::extract<char const *>(bpy::str(bar)) << std::endl;

结果打印出来:

Python ndarray : [-inf -inf -inf -inf -inf -inf -inf -inf -inf]

Python ndarray : [  0. -inf -inf -inf -inf -inf -inf -inf -inf]