如何将tqdm与map用于Dataframes

Question

我可以使用tqdm进度条和map函数来循环遍历数据帧/系列行吗？

具体而言,对于以下情况:

def example(x):
    x = x + 2
    return x

if __name__ == '__main__':
    dframe = pd.DataFrame([{'a':1, 'b': 1}, {'a':2, 'b': 2}, {'a':3, 'b': 3}])
    dframe['b'] = dframe['b'].map(example)

Answer 1

由于tqdm与pandas的集成,您可以使用progress_map函数而不是map函数.

注意:为此,您应该tqdm.pandas()在代码中添加一行.

试试这个:

from tqdm import tqdm

def example(x):
    x = x + 2
    return x

tqdm.pandas()  # <- added this line

if __name__ == '__main__':
    dframe = pd.DataFrame([{'a':1, 'b': 1}, {'a':2, 'b': 2}, {'a':3, 'b': 3}])
    dframe['b'] = dframe['b'].progress_map(example)  # <- progress_map here

这是文档参考:

(添加后tqdm.pandas())...您可以使用progress_apply替代apply和progress_map 取代map