如何以更Pythonic的方式通过函数传递类成员变量？

针对你的代码，这里有几个优化方向，能让代码更简洁、符合Python风格：

1. 修复初始化方法的错误 #

原代码中__init__计算var3和var4时未引用self，会触发NameError，首先要修正这个问题。

2. 替换冗余的var方法 #

Python内置的getattr()函数可以直接获取对象属性，完全不需要用一堆if判断来返回不同变量，这是最贴合Python风格的写法。

3. 增加参数合法性校验 #

为避免传入不存在的变量名导致报错，可以在类或函数中添加简单的校验逻辑，确保输入的变量名合法。

优化后的完整代码 #

import numpy as np
import matplotlib.pyplot as plt

class MyClass:
    # 定义合法变量名集合，方便后续校验
    VALID_VARS = {'var1', 'var2', 'var3', 'var4'}

    def __init__(self, input_path=None):
        if input_path is None:
            raise ValueError("必须提供输入文件路径")
        # 确保names=True读取结构化数据
        _tmp = np.genfromtxt(input_path, names=True)
        self.var1 = _tmp['f0']
        self.var2 = _tmp['f1']
        self.var3 = self.var1 + self.var2  # 修正：引用self.var1/var2
        self.var4 = self.var1 * self.var2  # 修正：引用self.var1/var2

def plot_MyClass(data1, data2, var='var1'):
    # 校验变量名合法性
    if var not in MyClass.VALID_VARS:
        raise ValueError(f"合法变量名只能是：{', '.join(MyClass.VALID_VARS)}")
    # 用getattr直接获取对象属性
    plt.plot(getattr(data1, var), getattr(data2, var), 'k-')

# 示例使用（注意文件名要加引号）
data1 = MyClass('file1.txt')
data2 = MyClass('file2.txt')
plot_MyClass(data1, data2, var='var3')
plt.show()

额外优化建议 #

• 如果想要更直观的变量访问方式，可以给MyClass添加__getitem__方法，这样就能像字典那样用data['var1']获取属性：

def __getitem__(self, key):
    if key not in self.VALID_VARS:
        raise KeyError(f"合法变量名只能是：{', '.join(self.VALID_VARS)}")
    return getattr(self, key)

之后在绘图函数中就可以写成data1[var]，可读性更强。

• 可以给plot_MyClass增加更多可选参数（比如线条样式、图例标签等），让函数更通用：

def plot_MyClass(data1, data2, var='var1', linestyle='k-', label=None):
    if var not in MyClass.VALID_VARS:
        raise ValueError(f"合法变量名只能是：{', '.join(MyClass.VALID_VARS)}")
    line, = plt.plot(getattr(data1, var), getattr(data2, var), linestyle)
    if label:
        line.set_label(label)
    plt.legend()

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