Python 多层 for 循环性能如何提高

V2EX = way to explore

V2EX 是一个关于分享和探索的地方

现在注册

已注册用户请登录

推荐学习书目

› Learn Python the Hard Way

Python Sites

› PyPI - Python Package Index

› http://diveintopython.org/toc/index.html

› Pocoo

值得关注的项目

› PyPy

› Celery

› Jinja2

› Read the Docs

› gevent

› pyenv

› virtualenv

› Stackless Python

› Beautiful Soup

› 结巴中文分词

› Green Unicorn

› Sentry

› Shovel

› Pyflakes

› pytest

Python 编程

› pep8 Checker

Styles

› PEP 8

› Google Python Style Guide

› Code Style from The Hitchhiker's Guide

这是一个创建于 789 天前的主题，其中的信息可能已经有所发展或是发生改变。

在实际业务中不得不遍历多个嵌套 for 循环，如何提高嵌套 for 循环的性能耗时太长了~

案例：比如一把武器 for 一个武器 in 武器库： for 弹药 in 弹夹： if 弹药是否属于该武器: for 弹药属性 in 弹药： # 判断弹药是否支持攻击某种物体

弹药

for

嵌套

循环

20 条回复 • 2022-09-30 14:40:08 +08:00

Building

2022-09-29 11:02:16 +08:00

提前用字典存好

eason1874

2022-09-29 11:22:26 +08:00

提前存好 +1

最朴素的方法往往是最有效的

dcty

2022-09-29 11:24:18 +08:00 via iPhone

空间换时间

SimonOne

2022-09-29 11:27:11 +08:00

请问 while 循环的间隔是 0.2s-0.3s 左右，有方法可以减少吗？

lookStupiToForce

2022-09-29 11:34:17 +08:00

把多层循环的内容根据逻辑改写难易丢 numpy 、丢 pandas 、丢数据库，最不济丢其他性能高的语言去处理呗

像你题目中说的，感觉用一个 df.apply 就能解决，前提是 merge 上必要的信息列

wxf666

2022-09-29 11:34:42 +08:00

同意 #1

1. 种类不多，你可以存成 `set[tuple[弹药, 该弹药能攻击的物体]]`：

```python
弹药能攻击的物体 = {
　(弹药 1, 物体 1),
　(弹药 1, 物体 2),
　(弹药 2, 物体 1),
　(弹药 2, 物体 3),
}
```

判断：`if (弹药 1, 物体 1) in 弹药能攻击的物体:`

2. 种类巨多，可能存成 `dict[弹药, set[该弹药能攻击的物体]]` 能省点内存？但比上面的慢

```python
弹药能攻击的物体 = {
　弹药 1: {物体 1, 物体 2},
　弹药 2: {物体 1, 物体 3},
}
```

判断：`if 物体 1 in 弹药能攻击的物体[弹药 1]:`

qW7bo2FbzbC0

2022-09-29 11:40:32 +08:00

用二维数据结构组织既有信息

也就是楼上说的用 map 或者 dict 来做

wcsjtu

2022-09-29 11:47:10 +08:00

思路一：修改逻辑，不要用 for 循环
思路二：实在要用 for ，改成 C/C++循环，可以使用 mypyc 编译该部分代码，也可以用 cython 编译。mypyc 有语法要求，性能稍微好一点。cython 可以直接编，性能比较差，但是还是远比 Python for 循环好

我们团队搞了一个专门为 Python 业务逻辑加速的编译器，性能吊打 cython/mypyc/pypy ，可惜不能开源.......