读代码的时候看到data, label = zip(*batch)
这样一句,很好奇它究竟实现了什么操作
利用zip(*)
命令将batch解压开,当有多个迭代器,希望它们能以相同索引一起输出时,会使用zip(*)
命令
class Student(object):
def __init__(self, score):
self.score = score
def __iter__(self):
return self # 对于迭代器来说,__iter__ 返回的是它自身self,也就是返回迭代器。
def __next__(self):
if self.score < 120:
self.score += 1
return [self.score]*5,2
else:
raise StopIteration()
test = Student(95)
print(isinstance(test, Iterable))
print(isinstance(test, Iterator))
ans=zip(*test)
data,label=list(ans)
print(data,label)
print(len(data),len(label))
输出
True
True
([96, 96, 96, 96, 96], [97, 97, 97, 97, 97], [98, 98, 98, 98, 98], [99, 99, 99, 99, 99], [100, 100, 100, 100, 100], [101, 101, 101, 101, 101], [102, 102, 102, 102, 102], [103, 103, 103, 103, 103], [104, 104, 104, 104, 104], [105, 105, 105, 105, 105], [106, 106, 106, 106, 106], [107, 107, 107, 107, 107], [108, 108, 108, 108, 108], [109, 109, 109, 109, 109], [110, 110, 110, 110, 110], [111, 111, 111, 111, 111], [112, 112, 112, 112, 112], [113, 113, 113, 113, 113], [114, 114, 114, 114, 114], [115, 115, 115, 115, 115], [116, 116, 116, 116, 116], [117, 117, 117, 117, 117], [118, 118, 118, 118, 118], [119, 119, 119, 119, 119], [120, 120, 120, 120, 120]) (2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2)
25 25
参考
https://blog.youkuaiyun.com/u010848594/article/details/106026597