python - 如果满足条件python从数组中减去一个数字

Question

我在 python 中的 if 条件中面临一个非常基本的问题。

current_img_list 数组的维度为 (500L,1)。如果这个数组中有数字 82459，我想将它减 1。

index = np.random.randint(0,num_train-1,500)
# shape of current_img_list is (500L,1)
# values in index array is randomized
current_img_list = train_data['img_list'][index]
# shape of current_img_list is (500L,1)

if (any(current_img_list) == 82459):
        i = np.where(current_img_list == 82459)
        final_list = i-1

变量说明——train_data是dict类型。它有 4 个元素。 img_list 是大小为 (215375L,1) 的元素之一。 num_train 的值为 215375，index 的大小为 (500L,1)

首先我不知道这个循环是否有效。我尝试了 all() 函数和 numpy.where() 函数但没有成功。其次，我想不出一种方法来直接从存储它的索引中从 82459 中减去 1，而不影响此数组中的其余值。

谢谢

Answer 1

在 Python 中遍历数组比让 numpy 的向量化运算符执行它们的操作要慢得多:

import numpy as np

num_train = 90000   # for example
index = np.random.randint(0,num_train-1,500)
index[index == 82459] -= 1