以下记录知识点：
1. Python: np.squeeze(train_set_y[:, index])
场景：在机器学习和深度学习中，通常算法的结果是可以表示向量的数组（即包含两对或以上的方括号形式[[]]
如果直接利用这个数组进行画图可能显示界面为空
我们可以利用squeeze（）函数将表示向量的数组转换为秩为1的数组，这样利用matplotlib库函数画图时，就可以正常的显示结果了。
https://blog.csdn.net/zenghaitao0128/article/details/78512715
2. zeros(shape, dtype=float, order = ‘C‘)