引言
通常，在 Python 中写循环（特别是多重循环）非常的慢，在文章 https://www.jb51.net/article/133807.htm中，我们的元胞自动机的状态更新函数 update_state 使用了两重循环，所以我们尝试用 Cython 重构该方法。
代码
我们在同文件夹下新建一个 update.pyx 文件，写入如下内容
import numpy as np
cimport numpy as np
cimport cython
DTYPE = np.float
ctypedef np.float_t DTYPE_t
def update_state(np.ndarray[DTYPE_t, ndim=2] cells):
return update_state_c(cells)
@cython.boundscheck(False)
@cython.wraparound(False)
cdef np.ndarray[DTYPE_t, ndim=2] update_state_c(np.ndarray[DTYPE_t, ndim=2] cells):
"""更新一次状态"""
cdef unsigned int i
cdef unsigned int j
cdef np.ndarray[DTYPE_t, ndim=2] buf = np.zeros((cells.shape[0], cells.shape[1]), dtype=DTYPE)
cdef DTYPE_t neighbor_num
for i in range(1, cells.shape[0] - 1):
for j in range(1, cells.shape[0] - 1):
# 计算该细胞周围的存活细胞数
neighbor_num = cells[i, j-1] + cells[i, j+1] + cells[i+1, j] + cells[i-1, j] + cells[i-1, j-1] + cells[i-1, j+1] + cells[i+1, j-1] + cells[i+1, j+1]
if neighbor_num == 3:
buf[i, j] = 1
elif neighbor_num == 2:
buf[i, j] = cells[i, j]
else:
buf[i, j] = 0
return buf
update_state_c 函数上的两个装饰器是用来关闭 Cython 的边界检查的。
在同文件下新建一个 setup.py 文件
import numpy as np
from distutils.core import setup
from Cython.Build import cythonize
setup(
name="Cython Update State",
ext_modules=cythonize("update.pyx"),
include_dirs=[np.get_include()]
)
因为在 Cython 文件中使用了 NumPy 的头文件，所以我们需要在 setup.py 将其包含进去。
执行 python setup.py build_ext --inplace 后，同文件夹下会生成一个 update.cp36-win_amd64.pyd 的文件，这就是编译好的 C 扩展。
我们修改原始的代码，首先在文件头部加入 import update as cupdate，然后修改更新方法如下
def update_state(self): """更新一次状态""" self.cells = cupdate.update_state(self.cells) self.timer += 1
将原方法名就改为 update_state_py 即可，运行脚本，无异常。
测速
我们编写一个方法来测试一下使用 Cython 可以带来多少速度的提升
def test_time():
import time
game = GameOfLife(cells_shape=(60, 60))
t1 = time.time()
for _ in range(300):
game.update_state()
t2 = time.time()
print("Cython Use Time:", t2 - t1)
del game
game = GameOfLife(cells_shape=(60, 60))
t1 = time.time()
for _ in range(300):
game.update_state_py()
t2 = time.time()
print("Native Python Use Time:", t2 - t1)
运行该方法，在我的电脑上输出如下
Cython Use Time: 0.007000446319580078
Native Python Use Time: 4.342248439788818
速度提升了 600 多倍。
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持。
免责声明：本站文章均来自网站采集或用户投稿，网站不提供任何软件下载或自行开发的软件！ 如有用户或公司发现本站内容信息存在侵权行为，请邮件告知！ 858582#qq.com
《魔兽世界》大逃杀！60人新游玩模式《强袭风暴》3月21日上线
暴雪近日发布了《魔兽世界》10.2.6 更新内容，新游玩模式《强袭风暴》即将于3月21 日在亚服上线，届时玩家将前往阿拉希高地展开一场 60 人大逃杀对战。
艾泽拉斯的冒险者已经征服了艾泽拉斯的大地及遥远的彼岸。他们在对抗世界上最致命的敌人时展现出过人的手腕，并且成功阻止终结宇宙等级的威胁。当他们在为即将于《魔兽世界》资料片《地心之战》中来袭的萨拉塔斯势力做战斗准备时，他们还需要在熟悉的阿拉希高地面对一个全新的敌人──那就是彼此。在《巨龙崛起》10.2.6 更新的《强袭风暴》中，玩家将会进入一个全新的海盗主题大逃杀式限时活动，其中包含极高的风险和史诗级的奖励。
《强袭风暴》不是普通的战场，作为一个独立于主游戏之外的活动，玩家可以用大逃杀的风格来体验《魔兽世界》，不分职业、不分装备（除了你在赛局中捡到的），光是技巧和战略的强弱之分就能决定出谁才是能坚持到最后的赢家。本次活动将会开放单人和双人模式，玩家在加入海盗主题的预赛大厅区域前，可以从强袭风暴角色画面新增好友。游玩游戏将可以累计名望轨迹，《巨龙崛起》和《魔兽世界：巫妖王之怒 经典版》的玩家都可以获得奖励。