Как использовать tqdm с картой для Dataframes

Question

Можно ли использовать индикатор выполнения tqdm с функцией карты для циклического перемещения по строкам данных / серии?

специально для следующего случая:

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 с пандами, вы можете использовать функцию 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