Python素数生成类优化：如何避免多次使用self参数？

问题背景 #

我创建了如下用于生成素数的Python类：

import math  # 补充原代码缺失的导入
class primeGen:
    def __init__(self, maxNum):
        self.allNum = [0]*(maxNum+1)
        self.allNum[0], self.allNum[1] = 1, 1
        self.primeNum = []
        for i in range(2, int(math.sqrt(maxNum))+1):
            if self.allNum[i] is 0:
                for j in range(i*i, len(self.allNum), i):
                    self.allNum[j] = 1
        for i in range(maxNum):
            if self.allNum[i] is 0:
                self.primeNum.append(i)
    def print(self):
        print(self.primeNum)

请问是否有办法在类的函数中避免多次使用self参数？例如，能否不用self.primeNum，而直接使用primeNum？

解答 #

好问题！在Python的类实例方法里，self确实是语法层面的硬性要求——它是当前实例的引用，用来区分实例属性和局部变量/全局变量。不过我们有几种实用的技巧可以减少self.的重复书写，同时保持代码的可读性和规范性：

1. 把常用实例属性赋值给局部变量

如果某个实例属性在方法里会被多次使用，你可以先把它赋值给一个简短的局部变量，之后直接用这个局部变量操作即可。这是最简单的变通方式，完全符合Python规范。

比如修改你的print方法：

def print(self):
    primes = self.primeNum
    print(primes)

在__init__方法里，也可以对高频使用的self.allNum做同样处理，同时建议用==代替is判断值相等（is是判断对象身份，虽然小整数在CPython中会缓存，但==更符合语义）：

def __init__(self, maxNum):
    self.allNum = [0]*(maxNum+1)
    nums = self.allNum
    nums[0], nums[1] = 1, 1
    self.primeNum = []
    primes = self.primeNum
    for i in range(2, int(math.sqrt(maxNum))+1):
        if nums[i] == 0:
            for j in range(i*i, len(nums), i):
                nums[j] = 1
    for i in range(maxNum):
        if nums[i] == 0:
            primes.append(i)

注意：局部变量是实例属性的引用，修改它会直接影响原实例属性，这通常是我们想要的效果。

2. 使用dataclasses简化属性定义（Python 3.7+）

如果你觉得在__init__里写一堆self.xxx = xxx很繁琐，可以用标准库的dataclasses自动生成初始化逻辑，虽然访问属性还是需要self.，但能大幅减少初始化代码里的self重复：

import math
from dataclasses import dataclass, field

@dataclass
class primeGen:
    maxNum: int
    # 用field标记不需要在__init__中传入的属性
    allNum: list = field(init=False)
    primeNum: list = field(init=False, default_factory=list)

    def __post_init__(self):
        # 初始化完成后执行的逻辑，不用再手动定义属性赋值
        self.allNum = [0]*(self.maxNum+1)
        nums = self.allNum
        nums[0], nums[1] = 1, 1
        primes = self.primeNum
        for i in range(2, int(math.sqrt(self.maxNum))+1):
            if nums[i] == 0:
                for j in range(i*i, len(nums), i):
                    nums[j] = 1
        for i in range(self.maxNum):
            if nums[i] == 0:
                primes.append(i)

    def print(self):
        print(self.primeNum)

3. 改用函数式风格（放弃类封装）

如果你的场景不需要维护实例状态，只是单纯生成素数，那完全可以放弃类，用普通函数实现，这样就彻底不需要self了：

import math

def generate_primes(maxNum):
    allNum = [0]*(maxNum+1)
    allNum[0], allNum[1] = 1, 1
    primeNum = []
    for i in range(2, int(math.sqrt(maxNum))+1):
        if allNum[i] == 0:
            for j in range(i*i, len(allNum), i):
                allNum[j] = 1
    for i in range(maxNum):
        if allNum[i] == 0:
            primeNum.append(i)
    return primeNum

# 使用示例
primes = generate_primes(100)
print(primes)

这种方式更简洁，但如果需要多次对同一素数集合进行操作（比如添加过滤逻辑、缓存结果），类的封装性会更有优势。

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