如何提取我的数组的随机行样本？

Question

我有一个名为 input_north 的 numpy 数组，其形状为 (823886, 13)

我想从这个数组中随机选择 82388 行并将它们存储在另一个名为真值的数组中。但是，我也想从原始 input_north 数组中删除随机采样的行。

这是我所做的，但它不能正常工作：

# first i make 82388 random indices between 1-823886 
indices = np.random.choice(823886, 82388)

# then I use the indices to create my truth array
truth = input_north[indicies, :]

# I then create a new array with the truth rows deleted
input = np.delete(input_north, indices, 0) 

当我这样做时，我会得到一个大小正确的真值数组 (82388, 13)，其中包含随机的行样本。但是，输入数组的形状为 (745496, 13)。这意味着并非所有提取到真值数组的行都已在输入数组中删除，因为它包含的行比我想要的多约 4000 行。如何从 inut_north 数据中提取行并删除已提取的行？

Answer 1

也许你可以这样尝试

import numpy as np

# create a array like:
# [1,1,1,1,...]
# [2,2,2,2,...]
# [3,3,3,3,...]
# [...]
# [N,N,N,N,...]

N = 100

data = np.ones((N, 13))
data[:] = data * np.arange(1,data.shape[0]+1)[:,None]
data = data.astype(np.int)

print(data)

# not perutate the array
np.random.shuffle(data)

print(data)

# now choose truth and input
split = 85
input = data[:split]
truth = data[split:]


print("input.shape",input.shape)
print("truth.shape",truth.shape)

input.shape (85, 13)
truth.shape (15, 13)