提供基于元胞自动机实现交通流的Python代码