Python中实现可索引自定义类属性的解决方案咨询

核心问题在于Numpy切片操作可能返回数组副本，直接对副本赋值无法同步修改原数组。通过实现代理类可以完美解决这个问题，既保留test.SecondRow[2] = 500这种直观语法，又确保所有修改直接作用于原数组。

解决方案代码 #

1. 实现列代理类 #

这个类负责封装原数组和目标列索引，直接通过__getitem__和__setitem__操作原数组的对应位置：

import numpy as np

class ColumnProxy:
    def __init__(self, arr, col_idx):
        self.arr = arr
        self.col_idx = col_idx
    
    def __getitem__(self, idx):
        # 直接返回原数组对应位置的值，无副本生成
        return self.arr[idx, self.col_idx]
    
    def __setitem__(self, idx, value):
        # 直接修改原数组对应位置，无需通过副本中转
        self.arr[idx, self.col_idx] = value

2. 修改Test类的属性实现 #

将原来返回切片的@property改为返回代理实例：

class Test:
    def __init__(self):
        self.arr = np.array([[1,2,3], [4,5,6], [7,8,9]])
    
    @property
    def SecondRow(self):
        # 返回代理对象，而非直接返回切片
        return ColumnProxy(self.arr, 1)

测试验证 #

test = Test()
print("修改前原数组：")
print(test.arr)
# 输出：
# [[1 2 3]
#  [4 5 6]
#  [7 8 9]]

# 使用直观的索引赋值语法修改
test.SecondRow[2] = 500

print("\n修改后原数组：")
print(test.arr)
# 输出：
# [[  1   2   3]
#  [  4   5   6]
#  [  7 500   9]]

扩展支持切片操作 #

代理类天然支持切片索引，无需额外修改：

test.SecondRow[0:2] = [100, 200]
print("\n切片修改后原数组：")
print(test.arr)
# 输出：
# [[ 100  200   3]
#  [  4   5   6]
#  [  7 500   9]]

优势说明 #

• 完全避免副本生成，节省内存开销，尤其适合大规模数组场景
• 保持test.SecondRow[idx]这种符合Python习惯的简洁语法
• 所有修改直接作用于原数组，彻底解决单元测试中频繁出现的修改失效问题

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