c++ - 编织内联类型转换(python)

Question

2013 年最美好的祝愿!

我正在使用 SciPy 的编织与我的一些 C++ 代码内联来转置巨大的矩阵(大约 200.000 x 15)。它就像一个魅力，但我有一个关于类型转换的问题:

我的输入矩阵是从文件中读取的，以逗号分隔等等，所以所有的条目都是字符串而不是 float ('0.551' 而不是 0.551)。这不会影响转置函数的工作方式，但后来我不得不将某些行转换为 numpy float 数组，所以我想知道是否可以在 C++ 代码中完成。让我用一些代码来解释:

def transpose(lines, N, x):
    code =  """
            py::list matrix;
            for(int i = 0; i < x; i++) {
                py::list line;
                if(i == 1) { continue; }
                for(int j = 0; j < N; j++) {
                    line.append(lines[j][i]);
                }
                matrix.append(line);
            }
            return_val = matrix;
            """
    return weave.inline(code, ['lines', 'N', 'x'])

matrix = [['0.5','0.1'],['0.2','0.2']]
matrixT = transpose(matrix, len(matrix), len(matrix[0]))
final_result = np.array(matrixT[0], dtype=float)

在示例中，我的小矩阵将被转置，我的示例结果将转置矩阵的第一行转换为 dtype float 的 numpy 数组。这可以用 C++ 代码代替吗？我试过使用 double x = (double) lines[j][i] 和类似的东西，但它在某种程度上不适用于附加到 py::list 对象。

Answer 1

以下可以完成您想要的所有事情:

def transpose(lines):
    code =  """
            for(int i = 0; i < x; i++) {
                for(int j = 0; j < N; j++) {
                    out[j + i * N] = atof(lines[j][i]);
                    // OUT2(i, j) = atof(lines[j][i]);
                }
            }
            """
    N = len(lines)
    x = len(lines[0])
    out = np.empty((x, N), dtype=np.float64)
    weave.inline(code, ['lines', 'N', 'x', 'out'])
    return out

>>> matrix = [['0.5', '0.1', '0.7'],['0.2','0.2', '0.4']]
>>> matrix
[['0.5', '0.1', '0.7'], ['0.2', '0.2', '0.4']]
>>> transpose(matrix)
array([[ 0.5,  0.2],
       [ 0.1,  0.2],
       [ 0.7,  0.4]])

除了经常忘记 ;在大约 6 年没有编写任何 C 之后，我很难弄清楚是什么 out正在进入 C++ 代码内部，最后它是一个指向数据本身的指针，而不是指向 PyArrayObject 的指针。作为 documentation状态。 weave 定义了两个可供使用的变量，out_array和 py_out , 属于 PyArrayObject* 类型和 PyObject*分别。

我留下了另一个版本的注释:weave automatically defines macros <VAR>1 , <VAR>2 , <VAR>3 , 和 <VAR>4访问相应维数的数组项。