如何避免在NumPy数组相关类中定义大量重复分贝转换属性？

针对你遇到的类中重复定义_db后缀属性导致代码冗余的问题，这里提供两种简洁的实现方式，无需扩展或包装numpy ndarray，同时保持代码可读性：

方案1：动态生成分贝属性 #

在类定义外部，通过循环为指定的线性属性自动生成对应的分贝属性。只需维护一个需要转换的字段列表，即可避免重复编写@property代码：

import numpy as np

def lin2db(x):
    return 10 * np.log10(x)

class Example:
    # 定义需要生成分贝属性的线性字段列表
    _db_target_fields = ['snr', 'eta']

    @property
    def snr(self):
        # 替换为你的实际计算逻辑
        return np.array([10, 100, 1000])
    
    @property
    def eta(self):
        # 替换为你的实际计算逻辑
        return np.array([0.1, 0.5, 0.9])

# 自动生成_db后缀的属性
for field in Example._db_target_fields:
    def create_db_property(field_name):
        @property
        def db_prop(self):
            return lin2db(getattr(self, field_name))
        return db_prop
    setattr(Example, f"{field}_db", create_db_property(field))

使用时依然保持原有调用方式：

obj = Example()
print(obj.snr)       # 输出线性值数组
print(obj.snr_db)    # 输出转换后的分贝值数组

方案2：使用类装饰器模块化实现 #

将生成分贝属性的逻辑封装为装饰器，与类定义分离，代码结构更清晰：

import numpy as np

def lin2db(x):
    return 10 * np.log10(x)

def add_db_properties(target_fields):
    """为类中的指定字段自动生成分贝属性"""
    def decorator(cls):
        for field in target_fields:
            def create_db_prop(field_name):
                @property
                def db_prop(self):
                    return lin2db(getattr(self, field_name))
                return db_prop
            setattr(cls, f"{field}_db", create_db_prop(field))
        return cls
    return decorator

# 应用装饰器，指定需要转换的字段
@add_db_properties(['snr', 'eta'])
class Example:
    @property
    def snr(self):
        return np.array([10, 100, 1000])
    
    @property
    def eta(self):
        return np.array([0.1, 0.5, 0.9])

方案优势 #

• 无需重复编写分贝转换的@property代码，只需维护字段列表
• 完全保留原生numpy ndarray的操作特性，没有额外复杂度
• 新增线性属性时，仅需将字段名添加到列表中，自动生成对应分贝属性

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