如何用pybind11绑定Array1D重载GetBuffer方法并解决Python赋值错误

你当前的绑定方式返回的原始指针被pybind11直接解析成了Python整数（对应指针的内存地址），而非可索引操作的数组对象，所以执行buff[i] = 1000时会触发类型错误——整数不支持元素赋值。

要让Python能识别并操作C++缓冲区，需要将指针包装成pybind11提供的数组/内存视图对象，而非直接返回原始指针。以下是两种可行的绑定方案：

方案一：用py::array_t包装（推荐，兼容numpy） #

通过lambda函数将GetBuffer()返回的指针包装成py::array_t，同时关联数组与Array1D对象的生命周期，避免悬空指针：

// 绑定const版本的GetBuffer
.def("GetBuffer", [](const Array1D<T>& self) {
    return py::array_t<const T>(
        {self.m_nSize},    // 数组长度（shape）
        {sizeof(T)},       // 元素 stride（字节数）
        self.GetBuffer(),  // 缓冲区起始指针
        py::cast(&self)    // 关联Array1D对象作为所有者，确保其存活时缓冲区有效
    );
}, py::return_value_policy::reference_internal)

// 绑定非const版本的GetBuffer
.def("GetBuffer", [](Array1D<T>& self) {
    return py::array_t<T>(
        {self.m_nSize},
        {sizeof(T)},
        self.GetBuffer(),
        py::cast(&self)
    );
}, py::return_value_policy::reference_internal);

在Python中调用时，返回的buff是numpy兼容的数组对象，可直接进行索引读写：

buff = lut.GetBuffer()
for i in range(lutSize):
    buff[i] = 1000

方案二：用py::memoryview包装 #

如果不需要numpy兼容性，可将指针包装成Python原生内存视图：

// 绑定const版本的GetBuffer（只读视图）
.def("GetBuffer", [](const Array1D<T>& self) {
    return py::memoryview::from_buffer(
        self.GetBuffer(),
        {self.m_nSize},
        {sizeof(T)},
        true  // 标记为只读
    );
}, py::return_value_policy::reference_internal)

// 绑定非const版本的GetBuffer（可写视图）
.def("GetBuffer", [](Array1D<T>& self) {
    return py::memoryview::from_buffer(
        self.GetBuffer(),
        {self.m_nSize},
        {sizeof(T)}
    );
}, py::return_value_policy::reference_internal);

• 确保Array1D的m_nSize成员能被绑定lambda访问，若m_nSize是protected成员，可将绑定代码声明为类的友元，或新增GetSize()公共方法来获取长度。
• 返回策略reference_internal确保Python不会拷贝缓冲区数据，直接操作原始内存；同时通过py::cast(&self)或内存视图的关联逻辑，保证Array1D对象存活期间缓冲区有效。

内容的提问来源于stack exchange，提问作者AAUDI