numpy:按索引拆分矩阵

Question

我有一个矩阵，我想根据行列表得到两个不同的矩阵。索引是随机创建的。我知道如何得到第一部分：

indexes = np.random.randint(low=0, high=num_rows size=splitsize)
part1 = data[indexes, :]

如何获取数据的其他部分？

Answer 1

也许您可以使用numpy.sediff1d来获取“其他”部分的索引，并使用它来索引矩阵。

In [24]: num_rows = 8

In [25]: indexes = [2,3,5]

In [26]: other = np.setdiff1d(np.arange(num_rows), indexes)

In [27]: other
Out[27]: array([0, 1, 4, 6, 7])