基于astropy.units自定义MJy/steradian单位及带参转换问询

一、自定义MJy/steradian单位 #

Astropy本身已经内置了u.MJy（兆 Jansky）和u.steradian（球面度），直接组合就能得到目标单位，也可以注册为自定义别名方便复用：

from astropy import units as u

# 方式1：直接组合已有单位（最便捷）
mjy_per_sr = u.MJy / u.steradian

# 方式2：注册为自定义单位（可选，适合流水线统一调用）
u.def_unit('MJy_per_steradian', u.MJy / u.steradian, doc="Mega Jansky per steradian")

二、带像素尺寸的单位转换（MJy/sr → Jy/pixel） #

Astropy支持通过**等价关系（equivalencies）**实现带参数的单位转换，完全不用自己写转换类，还能复用Astropy原生Quantity的所有特性。核心逻辑是：Jy/pixel = MJy/sr × 像素对应的立体角，步骤如下：

1. 定义转换等价关系 #

from astropy import units as u

def pixel_brightness_equivalency(pixel_size_arcsec):
    # 把弧秒像素尺寸转为立体角：1平方弧秒 = (π/180/3600)² 球面度
    pixel_solid_angle = (pixel_size_arcsec * u.arcsec)**2.to(u.steradian)
    # 建立双向转换规则：MJy/sr ↔ Jy/pixel
    return [
        (u.MJy / u.steradian, u.Jy / u.pixel,
         # MJy/sr → Jy/pixel：1 MJy = 1e6 Jy，乘以像素立体角
         lambda x: x * 1e6 * pixel_solid_angle.value,
         # Jy/pixel → MJy/sr：反向计算
         lambda x: x / (1e6 * pixel_solid_angle.value))
    ]

2. 实际转换示例 #

# 定义原始面亮度值
brightness = 10.0 * (u.MJy / u.steradian)
# 像素尺寸（单位：弧秒）
pixel_size = 0.25

# 开启等价关系后执行转换
with u.set_enabled_equivalencies(pixel_brightness_equivalency(pixel_size)):
    jy_per_pixel = brightness.to(u.Jy / u.pixel)

print(f"转换结果：{jy_per_pixel.value:.6f} {jy_per_pixel.unit}")
# 输出：转换结果：0.000147 Jy / pixel

三、流水线场景优化 #

如果流水线中频繁用到不同像素尺寸的转换，可以封装成工具函数，避免重复代码：

class PipelineUnitTools:
    @staticmethod
    def brightness_to_jy_per_pixel(brightness_quantity, pixel_size_arcsec):
        with u.set_enabled_equivalencies(pixel_brightness_equivalency(pixel_size_arcsec)):
            return brightness_quantity.to(u.Jy / u.pixel)

# 流水线调用示例
pipeline_data = 15.0 * u.MJy_per_steradian
result = PipelineUnitTools.brightness_to_jy_per_pixel(pipeline_data, 0.5)
print(result)  # 输出：0.000588 Jy / pixel

优势对比 #

相比自定义转换类，这种方式的好处：

• 完全兼容Astropy的Quantity类，支持数组运算、单位校验、与其他天文工具联动
• 支持双向转换（Jy/pixel转MJy/sr）
• 等价关系可灵活开关，适合批量数据处理

内容的提问来源于stack exchange，提问作者user19416164