pandas 使用技巧

# 根据定好的columns去给数据集填值
# 数据集中与定好的columns中匹配的保留，缺失的根据fill_value的值进行填充，多余的忽略
a = pd.DataFrame({'haha': range(5), 'didi':range(20, 5), 'cc': range(40, 5)})
t_col = ['haha', 'fafa', 'didi', 'gg']
a.reindex(columns=t_col, fill_value=0)
输出：
	index haha	fafa	didi	gg
	0	0	0	20	0
	1	1	0	21	0
	2	2	0	22	0
	3	3	0	23	0
	4	4	0	24	0
	5	5	0	25	0

one-hot转类别值
[np.argmax(one_hot) for one_hot in y_predict]