参考for循环中的数组

时间:2018-03-01 20:49:40

标签: python python-3.x numpy for-loop

假设我有3个Numpy数组,其中包含以下名称的浮点数:一两三。

是否可以通过以下方式在for循环中引用它们:

list=one two three

for arr in list:
    arr=arr[2,1]+1

上面的例子显然不起作用,但我想知道是否有办法做到这一点?

2 个答案:

答案 0 :(得分:1)

您需要指定您要保存值的位置:

arr[2,1] = arr[2,1] + 1

或只是:

arr[2,1] += 1

所以整个代码变成了:

import numpy as np
one = np.arange(6).reshape(3, 2)
two = np.arange(10).reshape(5, 2)
arrays = [one, two]
for arr in arrays:
    arr[2, 1] += 1

答案 1 :(得分:0)

我认为如果你将列表变量列为一个列表,它会起作用:

list=[one, two, three]
for arr in list:
    arr=arr[2,1]+1