当数据帧初始化时，如何创建进度条？

Question

当我将.csv文件加载到数据帧中时，每次创建一个新行时，我都希望获得行数：

def file_len(fname):
    with open(fname) as f:
        for i, l in enumerate(f):
            pass
    return i + 1

csv_path = "C:/...."
max_length = file_len(csv_path)

data = read_csv(csv_path, sep=';', encoding='utf-8')

使用这段代码，我得到了最大的行数，但我不知道如何在每次创建数据帧时得到数据帧中的行数。我想用它们来制作一个0-100%的进度条

Answer 1

您不能这样做--您必须修改read_csv函数以及Pandas中的其他函数。

编辑：

现在似乎可以用chunksize=rows_number来完成。

import pandas as pd

from StringIO import StringIO

data = """A,B,C
foo,1,2,3
bar,4,5,6
baz,7,8,9
"""

reader = pd.read_csv(StringIO(data), chunksize=1)

for x in reader:
    print x
    print '--- next data ---'

     A  B  C
foo  1  2  3
--- next data ---
     A  B  C
bar  4  5  6
--- next data ---
     A  B  C
baz  7  8  9
--- next data ---