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
最后的结果显示如下: