pandas读取CSV文件创建数据生成器循环迭代

阅读量：

背景

数据集的标签信息以csv文件形式存储，包含三个字段column_hander = [‘id’, ‘boneage’, ‘male’]，需自行构建对应的数据集。由于文件体积较大，建议采用数据生成器的方式实现数据的逐批加载与迭代处理。

实现模板

通过Pandas库提供的pandas.read_csv方法，能够实现对CSV格式文件的数据加载操作，同时借助iter函数可将生成的DataFrame对象转化为迭代器形式，从而在每次调用next方法时逐行获取数据内容。

复制代码

    import pandas as pd
     
    # 读取CSV文件到DataFrame
    df = pd.read_csv('your_file.csv')
     
    # 将DataFrame转换为迭代器
    data_iter = iter(df.itertuples(index=False, name=None))
     
    # 使用next获取每行数据
    try:
    while True:
        row = next(data_iter)
        print(row)
    except StopIteration:
    print("迭代完成")

全部评论 (0)

还没有任何评论哟~