遍历一个numpy数组

koko 发表于 Dev

尺寸

我有一个5 x 10的数组，如果随机数大于0.9，我想翻转一下。但是，这仅适用于数组的第一行，而不能到达第二行和后续行。我用3和4替换了这些位，因此我可以轻松查看是否发生翻转。我得到的结果看起来像这样。

[[3 1 1 1 4 1 3 1 0 1]
 [1 1 0 0 1 0 1 1 1 0]
 [1 0 1 0 1 0 1 1 1 1]
 [0 0 1 0 1 1 0 1 1 1]
 [0 1 1 0 0 0 0 1 1 1]]

请帮助我找出我错了。


from random import random
RM =  np.random.randint(0,2, size=(5,10))
print(RM)
for k in range(0, RM.shape[0]):
    for j in range(0, RM.shape[1]):
        A  = random()
        if A > 0.9:
            if RM[k,j] == 0:
                np.put(RM, [k,j], [3]) 
                print("k1",k)
                print("j1", j)
            else:
                np.put(RM, [k,j], [4])  
                print("k2", k)
        else: 
            continue

print(RM)

翘曲

查看np.put的文档

numpy.put(a, ind, v, mode='raise')[source]
Replaces specified elements of an array with given values.

在示例下：

a = np.arange(5)
np.put(a, [0, 2], [-44, -55])
a
array([-44,   1, -55,   3,   4])

因此，如果您向该函数提供列表，它将替换扁平化数组中的多个值。

为了使循环正常工作，只需将值分配给数组即可工作：

from random import random
RM =  np.random.randint(0,2, size=(5,10))
print(RM)
for k in range(0, RM.shape[0]):
    for j in range(0, RM.shape[1]):
        A  = random()
        if A > 0.9:
            if RM[k,j] == 0:
                RM[k,j]=3 
                print("k1",k)
                print("j1", j)
            else:
                RM[k,j] =4  
                print("k2", k)
        else: 
            continue

本文收集自互联网，转载请注明来源。

如有侵权，请联系 [email protected] 删除。