这段时间在打一个机器学习比赛，遇到分层抽样的需求。在网上查了一下，虽然也有前辈用pandas写出来过，但代码有很多冗余，于是我又重写了一个超简洁的版本。注释应该还算清晰，各位可以自取。seed随机抽取与打乱顺序时用到的随机数种子，默认为None。当输入一个数字的时候，输出的样本及样本顺序都会固定。返回的是训练样本和测试样本，两个样本不重叠，且合在一起包含了全部样本。sample_data原始数据，

MATLAB基于元胞自动机模型的传染病扩散模型基本思路：地图矩阵可以分为两类，一类是健康人矩阵，一类是感染者矩阵。健康人感染后则落入感染者矩阵，感染者康复后则升上健康人矩阵通过建立三维矩阵Track（在x轴和y轴之上加入时间轴形成三维）来追踪已感染的人的感染时间，从而判定康复。通过设置NewMap和NewPatientMap两个过渡矩阵来更新每次的随机移动，防止先更新的人在遍历过程中再次被选中导致