时间戳

首页 > 解决方案 > 如何简洁地获取 Sympy 表达式的单位

python - 如何简洁地获取 Sympy 表达式的单位

问题描述

我正在使用具有单位的 Sympy 表达式。这是一个简单的例子：

from sympy.physics.units import newton, meter
atmosphere = 101325. * newton / (meter **2)

在上面的例子中，是否有一种简洁的方法来获取单位，即newton / (meter **2)从变量中获取atmosphere？

标签： pythonsympy

解决方案

你的方法很好，但我会is_number用has(Quantity)

from sympy.physics.units import Quantity,
def unit(expr):
    return expr.subs({x: 1 for x in expr.args if not x.has(Quantity)})

然后它也适用于象征性的数量，比如

g = symbols('g')
acc = g*meter/second**2
unit(acc)  # meter/second**2

推荐阅读