python定义整数数组_python – TypeError：只能将整数标量数组转...

我想编写一个函数,从训练集中随机选取元素,但是基于提供的bin概率.我将设置索引除以11个bin,然后为它们创建自定义概率.

bin_probs = [0.5, 0.3, 0.15, 0.04, 0.0025, 0.0025, 0.001, 0.001, 0.001, 0.001, 0.001]

X_train = list(range(2000000))

train_probs = bin_probs * int(len(X_train) / len(bin_probs)) # extend probabilities across bin elements

train_probs.extend([0.001]*(len(X_train) - len(train_probs))) # a small fix to match number of elements

train_probs = train_probs/np.sum(train_probs) # normalize

indices = np.random.choice(range(len(X_train)), replace=False, size=50000, p=train_probs)

out_images = X_train[indices.astype(int)] # this is where I get the error

我收到以下错误：

TypeError: only integer scalar arrays can be converted to a scalar index with 1D numpy indices array

我发现这很奇怪,因为我已经检查了我创建的索引数组,它是1-D,它是整数,它是标量.

我错过了什么？

注意：我试图用astype(int)传递索引.同样的错误.