python - NumPy 对于涉及数组各个元素的算法的性能

Question

我对 NumPy 的性能感兴趣，当涉及到检查元素及其从属关系(例如相邻元素)的条件是否为 True 并根据条件分配值的算法时。

一个例子可能是:(我现在编的)

• 我随机生成一个由 1 和 0 组成的二维数组。
  • 然后检查数组的第一个元素是否与其相邻元素相同。
  • 如果相似元素占大多数，我会切换(0 -> 1 或 1 -> 0)该特定元素。
• 然后我将继续处理下一个元素。

我猜想 NumPy 的这种逐元素条件和逐元素操作相当慢，有没有办法可以提高性能？

例如，创建 dbool 类型的数组并调整代码，会有帮助吗？

提前致谢。

Answer 1

也许http://www.scipy.org/Cookbook/GameOfLifeStrides对你有帮助。