小编Tap*_*ala的帖子

sympy 的高级索引?

使用 numpy,我可以从带有整数列表的数组中选择任意一组项目:

>>> import numpy as np
>>> a = np.array([1,2,3])
>>> a[[0,2]]
array([1, 3])
Run Code Online (Sandbox Code Playgroud)

这似乎不适用于 sympy 矩阵,因为代码:

>>> import sympy as sp
>>> b = sp.Matrix([1,2,3])
>>> b[[0,2]]
Run Code Online (Sandbox Code Playgroud)

结果出现错误消息:

**Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/usr/lib/python2.7/dist-packages/sympy/matrices/dense.py", line 94, in __getitem__
    return self._mat[a2idx(key)]
  File "/usr/lib/python2.7/dist-packages/sympy/matrices/matrices.py", line 4120, in a2idx
    raise IndexError("Invalid index a[%r]" % (j, ))
IndexError: Invalid index a[[0, 2]]
Run Code Online (Sandbox Code Playgroud)

我的问题是是否有办法在 sympy 中做到这一点?

python numpy sympy

7
推荐指数
1
解决办法
1055
查看次数

标签 统计

numpy ×1

python ×1

sympy ×1