如何简明地得到符号表达式的单位

问题描述

我正在使用具有单位的符号表达式。下面是一个简单的例子：

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

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

---

解决方案

您的方法很好，但我会将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