Pandas to_csv 进度条与 tqdm

pandas

1个回答

写回答

yanyancup

2025-07-09 16:55

+ 关注

Pandas
Pandas

使用Pandas库的to_csv函数可以方便地将数据保存为CSV文件格式。然而,如果处理的数据量较大,保存过程可能会花费较长的时间,这时候可以使用tqdm库来添加进度条,以便更直观地监视保存的进度。

在进行大数据量的保存操作时,进度条是非常有用的。通过显示进度条,我们可以清楚地看到保存的进度,从而更好地掌握操作的耗时和剩余时间。

下面,我们将结合一个案例来演示如何使用Pandas的to_csv函数和tqdm库来保存数据并显示进度条。

首先,我们需要导入所需的库:

Python

import Pandas as pd

from tqdm import tqdm

接下来,我们定义一个示例数据集,并将其保存为CSV文件:

Python

data = pd.DataFrame({'姓名': ['张三', '李四', '王五'],

'年龄': [20, 25, 30],

'性别': ['男', '女', '男']})

with tqdm(Total=len(data.index), desc='保存进度') as pbar:

data.to_csv('example.csv', index=False)

pbar.update(len(data.index))

在上述代码中,我们创建了一个包含姓名、年龄和性别的示例数据集,并使用to_csv函数将其保存为名为example.csv的CSV文件。使用tqdm库的with语句,我们创建了一个进度条对象pbar,并设置总的迭代次数为数据集的行数。在保存数据之前,我们调用pbar.update(len(data.index))来更新进度条的状态,表示已经完成了所有的保存操作。

通过运行上述代码,我们可以看到一个名为"保存进度"的进度条,随着保存的进行,进度条将逐渐增长,直到保存完成。

案例代码:

Python

import Pandas as pd

from tqdm import tqdm

data = pd.DataFrame({'姓名': ['张三', '李四', '王五'],

'年龄': [20, 25, 30],

'性别': ['男', '女', '男']})

with tqdm(Total=len(data.index), desc='保存进度') as pbar:

data.to_csv('example.csv', index=False)

pbar.update(len(data.index))

通过以上案例代码,我们可以在保存大数据量时,添加进度条来直观地展示保存的进度。这样可以提高操作的效率,并更好地掌握操作的进展情况。

举报有用(4)分享收藏

Copyright © 2025 IZhiDa.com All Rights Reserved.

知答 版权所有 粤ICP备2023042255号