6.4 使用numpy解决扩散问题