Sympy:具有大型数组输入的Lambdify函数

Question

我在定义一个需要一个大数组的函数时遇到了问题。复制相同问题的简化代码是：

from sympy import *
def fun(x):
    f = []
    for i,x_i in enumerate(x):
        f.append(x_i**i)
    return Matrix(f)
N = 256
x = Matrix([symbols("x_%s"%i) for i in range(N)])
fun_lam = lambdify((x,),fun(x))

这会给出以下错误：

Traceback (most recent call last):
  File "bin/problem-lambdify.py", line 13, in <module>
    fun_lam = lambdify((x,),fun(x))
  File ".../env/lib/python3.4/site-packages/sympy-1.1rc1-py3.4.egg/sympy/utilities/lambdify.py", line 434, in lambdify
    func = eval(lstr, namespace)
  File "<string>", line 1
SyntaxError: more than 255 arguments

我认为，在某些时候，渐近会使论点变得扁平，从而导致这个问题。

我想不出一个好办法来解决这个问题。

Answer 1

显然，Python3.7(将于2018年发布)将removing这一限制。我知道这现在对您没有帮助(除非您想在Python的开发版本上运行)，但至少对未来有希望。

Answer 2

我吃了很多苦头，但我找到了解决办法。下面是一个简单的例子。

import sympy as sym
import numpy as np

A = sym.MatrixSymbol('A', 1, 260)
f = np.prod(A)
func = sym.lambdify(A, f)
aa = np.ones((1, 260))
print(func(aa))

诀窍是不创建符号数组，而是创建一个矩阵符号。但是我非常期待通过传递lambdify来实现Sympy中的普通数组。