使用帶有 ctypes 的 C function 計算 Python 中的熵

Question

我正在嘗試使用 C function 使用 ctypes 庫計算 Python 腳本中的 Sahnnon 熵。

我查閱了一些教程，據我了解，C代碼可以是這樣的（我叫“entropy.c”：

#include <math.h>
#include <stdio.h>

float entropy(float *p, int n)
{

   int i;
   float ent;

   ent = 0.0;
   for(i=0; i< n; i++)
   {
      if(p[i] > 0.0)
      {
         ent += -p[i] * (double) log( (double) p[i]);
      }
    }

   return ent;
}

在 C 代碼中，有一個變量數組 (p) 表示必須計算熵的分布。

現在我以通常的方式將其轉換為“共享對象”文件：

$ gcc -fPIC -shared -o entropy.so entropy.c

因此，據我了解，Python 中的代碼必須如下所示：

import ctypes

so_file="./entropy.so"

entropy=ctypes.CDLL(so_file)


n=ctypes.c_int(2)
p=(ctypes.c_float * 2)(0.25,0.75)

print(entropy.entropy(p,n))

但是，無論 p 是什么，我總是得到以下結果：

$ python entropy.py
$ 2

我試圖找到解決方案，但找不到。 所以問題是我在這里做錯了什么？

謝謝你的幫助。

Answer 1

您沒有告訴 Python 返回類型，所以它猜測它是int ，錯誤地，因此從錯誤的位置讀取返回值（假設您在 x86-64 上，它從eax而不是xmm0 ）並解釋它錯了。 在print(entropy.entropy(p,n))之前執行entropy.entropy.restype = ctypes.c_float 。