numpy:按索引拆分矩阵

时间:2015-10-10 16:36:26

标签: python numpy

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

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

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

1 个答案:

答案 0 :(得分: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])