【琐碎】如何理解zip(*batch)

最新推荐文章于 2024-09-08 02:25:32 发布

原创

最新推荐文章于 2024-09-08 02:25:32 发布 · 2.4k 阅读

9 ·

CC 4.0 BY-SA版权

文章标签：

#tensorflow #人工智能 #python

本文解析了Python中`zip(*batch)`命令如何将batch中的迭代器按索引打包成元组对，用于同时处理多个数据源。通过实例展示了如何创建自定义迭代器生成数据和标签，最后输出了数据和标签的列表以及长度。

读代码的时候看到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