当 numpy 数组被 C 函数改变时，魔法正在发生

Question

我正在尝试将 numpy 数组传递给我想要更改数组值的 C 函数。但我必须将索引乘以 2 才能使其正常工作。为奇数索引分配一个值（让它为i）会导致将一个垃圾分配给索引(i-1)/2。这是源代码。

C:

#include "assign_array.h"

void assign(int *arr, int i) {
    arr[i] = 2017;
}

Python：

import ctypes, numpy
lib2 = ctypes.cdll.LoadLibrary('./assign_array.so')
for i in range(5):
    array = numpy.zeros(8, dtype = int)
    ptr = array.ctypes.data_as(ctypes.POINTER(ctypes.c_int))
    print('i = {}:'.format(i))
    lib2.assign(ptr, i)
    print(array)

Screenshot

为什么会这样？

Answer 1

这不是完全垃圾：2047 是 0x7e1，8662949036032 是 0x7E100000000。看起来好像您正在混合不同的 C 编译器（或提供给 C 编译器的标志），因此 Python 认为 int 是 64 位，但您的 C 编译器认为 int 是 32 位。虽然事实上，Python 中的 int 不能保证与 C 编译器中的 int 匹配；它可能更典型地匹配long。

根据the numpy documentation，dtype可以指定为numpy.int32，专门指一个32位的int。这可能是处理此问题的最简单方法，或者您当然可以更改 C 代码以使用 long（假设您的 C 编译器设置为 64 位 longs）。