如何在初始化DataFrame時創建進度條？

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 。

僅使用iterator=True對我不起作用 - 或者可能需要更多行。

謝謝 Jeff

試試這個

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 ---