Question

我创建了一个此类，该类返回矢量的非零值。

class SparVec:

    def __init__(self,length):
        self.val={}
        self.len=length

    def __len__(self):
        return self.len

    def __getitem__(self, index):
        return self.val.get(index,0)
    def __setitem__(self,index,value):
        if value !=0:
             self.val[index]=value
        self.len=max(index,self.len)
    # Adding vectors
    def __add__(self, other):
        # Add Two vectors
        length=self.len
        result=SparVec(length)
        for index in self.val:
            result[index]=self[index]+other[index]
        for index in other.val:
            if index not in self.val:
                result[index]=other[index]
        return result

    def nonzeros(self):
        return 'Sparse Vector {}'.format(self.val)

if __name__ == '__main__':

    a = SparVec(4)
    a[2] = 9.2
    a[0] = -1
    a[3] = 0
    print a
    print a.nonzeros()

    b = SparVec(5)
    b[1] = 1
    print b
    print b.nonzeros()

    c = a+b
    print c
    print c.nonzeros()

到目前为止，一切都很好。我可以通过nonzeros函数获得所需的值。问题是当我打印a或b时如何具有以下输出。提示：无需创建一个将所有项目都存储为零的新列表，然后进行打印。

print a
[0]=-1 [1]=0 [2]=9.2 [3]=0

Answer 1

您可以将str.join与生成器表达式一起使用，该表达式在稀疏矢量的长度上进行迭代并输出默认值为0的键值对：

def __str__(self):
    return ' '.join('[{}]={}'.format(i, self.val.get(i, 0)) for i in range(self.len))

如何在课堂上实现str函数？

1 个答案: