在python中命名物理量

时间:2010-11-19 17:07:34

标签: python coding-style naming readability scientific-computing

我想为我的模拟代码中使用的物理/数学量建立一个好的命名方案。请考虑以下示例:

from math import *

class GaussianBeamIntensity(object):
    """
    Optical intensity profile of a Gaussian laser beam.
    """

    def __init__(self, intensity_at_waist_center, waist_radius, wavelength):
        """
        Arguments:

        *intensity_at_waist_center*: The optical intensity of the beam at the
            center of its waist in W/m^2 units.

        *waist_radius*: The radius of the beam waist in meters.

        *wavelength*: The wavelength of the laser beam in meters.

        """

        self.intensity_at_waist_center = intensity_at_waist_center
        self.waist_radius = waist_radius
        self.wavelength = wavelength
        self._calculate_auxiliary_quantities()

    def _calculate_auxiliary_quantities(self):
        # Shorthand notation
        w_0, lambda_ = self.waist_radius, self.wavelength

        self.rayleigh_range = pi * w_0**2 / lambda_
        # Generally some more quantities could follow

    def __call__(self, rho, z):
        """
        Arguments:

        *rho*, *z*: Cylindrical coordinates of a spatial point.
        """
        # Shorthand notation
        I_0, w_0 = self.intensity_at_waist_center, self.waist_radius
        z_R = self.rayleigh_range

        w_z = w_0 * sqrt(1.0 + (z / z_R)**2)
        I = I_0 * (w_0 / w_z)**2 * exp(-2.0 * rho**2 / w_z**2)
        return I

为了在可读性简洁符号之间进行平衡,您会为物理属性(属性,函数参数等)建议一致的命名方案(公式保持相对短)?你能否完善上面的例子?或者提出更好的方案?

遵循PEP8的指导方针会很高兴,记住“愚蠢的一致性是小头脑的大人物”。在遵守传统的80行字符长度限制时,似乎很难坚持使用描述性名称。

提前谢谢!

2 个答案:

答案 0 :(得分:4)

我认为你已经找到了良好的平衡。表达名称很重要,所以我完全同意使用 wavelenght 而不是lambda作为类属性。通过这种方式,界面保持清晰和富有表现力。

在一个长公式中,lambda_是速记符号的好选择,因为这是光学中波长的普遍接受和广泛使用的符号。我认为当你实现一个公式时,你想要做的就是尽可能接近你在一张纸上写的方程式(或者它们出现在文章中等)。

简而言之:保持界面表达,公式简​​短。

答案 1 :(得分:0)

使用Python3,您可以使用实际符号λ作为变量名。

我期待编写如下代码:

from math import pi as π

sphere_volume = lambda r : 4/3 * π * r**3