pandas中使用tqdm

pandas中使用tqdm

一般使用python的tqdm，可以查看到处理数据的进度，但是如果想要把tqdm应用到pandas中，则需要改变一下用法。

同样需要先导入tqdm模块：

from tqdm import tqdm

使用tqdm.pandas，同时input_data为Dataframe结构，因此需要使用input_data.progress_apply来实现apply操作：

def get_feature(input_data):
    """
    Get feature data
    """
    feature_data = np.zeros([len(input_data), hp.feature_len])
    error_sample_index = []

    tqdm.pandas(desc="get feature in input_data")
    input_data.progress_apply(lambda x: apply_feature(x, feature_data, error_sample_index), axis=1)

    if len(error_sample_index) != 0:
        feature_data = np.delete(feature_data, error_sample_index, axis=0)
    return feature_data

最后的结果显示如下：