python-3.x - 同情 pprint sin(a_n) 和 s(a_n)
问题描述
为了节省空间,我正在尝试使用带有第一个字母而不是缩写词的三角函数来对 pprint 矩阵表示同情。所以 sin(a_1) 看起来像 sa_1。现在我正在打印到文本然后运行查找和替换。(我是编程新手。)到目前为止,这是行不通的:
from sympy import sin as s
from sympy import cos as c
# declaring symbolic variables:
sin, cos, = sym.symbols('s, c')
#An example Matrix
T = Matrix([[c(theta), -s(theta), 0, a],
[s(theta) * c(alpha), c(theta) * c(alpha), -s(alpha), -s(alpha) * d],
[s(theta) * s(alpha), c(theta) * s(alpha), c(alpha), c(alpha) * d],
[0, 0, 0, 1]])
#T04 was a sympy symbol matrix solution
T_000 = str(print_latex(T04))
T_000 = str(T04)
T_000 = T_000.replace('sin', 's')
T_000 = T_000.replace('cos', 'c')
print('T000\n')
pprint(T_000)
T_001 = Matrix([[(-s(theta_1) * s(theta_2) + c(theta_1) * c(theta_2)) * c(theta_3) + (
-s(theta_1) * c(theta_2) - s(theta_2) * c(theta_1)) * s(theta_3),
-(-s(theta_1) * s(theta_2) + c(theta_1) * c(theta_2)) * s(theta_3) + (
-s(theta_1) * c(theta_2) - s(theta_2) * c(theta_1)) * c(theta_3), 0,
L_1 * c(theta_1) + L_2 * (-s(theta_1) * s(theta_2) + c(theta_1) * c(theta_2))], [
(-s(theta_1) * s(theta_2) + c(theta_1) * c(theta_2)) * s(theta_3) + (
s(theta_1) * c(theta_2) + s(theta_2) * c(theta_1)) * c(theta_3),
(-s(theta_1) * s(theta_2) + c(theta_1) * c(theta_2)) * c(theta_3) - (
s(theta_1) * c(theta_2) + s(theta_2) * c(theta_1)) * s(theta_3), 0,
L_1 * s(theta_1) + L_2 * (s(theta_1) * c(theta_2)
+ s(theta_2) * c(theta_1))], [0, 0, 1, d_4], [0, 0, 0, 1]])
print('\nT001\n')
pprint(T_001)
T_000.replace(sin, s)
print('T000\n', T_000)
它总是以完整的“sin”和“cos”名称打印。
解决方案
有两种常用的方法来替换表达式中的函数(在这种情况下,表达式是矩阵T
)。一是使用replace
功能,二是使用subs
功能。两者中的任何一个都可用于实现将 sin/cos 替换为其他函数的相同目标。
在下面的代码中,我们将函数s
(它是 的别名sin
)替换为未定义的函数s1
(其符号名称为s
)。和c
也是如此c1
。
from sympy import *
# s,c act as aliases to sin,cos
s = sin
c = cos
# the intended short-hand functions used to replace sin/cos later on
s1 = Function('s')
c1 = Function('c')
a,d,theta,alpha = symbols('a d \\theta \\alpha')
T = Matrix([
[c(theta), -s(theta), 0, a],
[s(theta) * c(alpha), c(theta) * c(alpha), -s(alpha), -s(alpha) * d],
[s(theta) * s(alpha), c(theta) * s(alpha), c(alpha), c(alpha) * d],
[0, 0, 0, 1]
])
T_1 = T.replace(s,s1).replace(c,c1)
T_2 = T.subs({s:s1,c:c1})
display(T_1)
display(T_2)
输出:
这篇文章中使用的代码也可以在这里找到。
推荐阅读
- javascript - 使用浏览器中的 create-react-app 读取和写入文本文件
- ios - 如何在for循环中按顺序运行函数
- python - 在函数中导入模块会产生问题吗?
- javascript - 表格排序器问题
- c# - 如何在 UWP 的主窗口上打开 NewWindow 并禁用后台 Window(MainWindow) 交互?
- ruby-on-rails - 救援后台作业无法从 AWS Secrets Manager 加载配置值
- reactjs - React-如何在页面刷新后使状态持续存在?
- c++ - 如何制作倒数猜谜游戏的功能?
- python - pd.read_excel 并不总是读取所有列
- sql - 从 spark 函数调用转换为 SQL