python - 编写 python for 循环的更短方法

Question

在 Python 中有更短的写法吗？这里 d 是一个 python 字典，a 是一个 numpy 数组。

i = 0
for b in a:
    d[b] = a[:,i]
    i += 1

`

Answer 1

您可以使用enumerate 迭代器:

for i, b in enumerate(a):
    d[b] = a[:,i]

因为你要进入一个字典，你可以这样做:

d.update((b, a[:,i]) for i, b in enumerate(a))