我有一个矩阵,我希望根据行列表得到两个不同的矩阵。索引是随机创建的。我知道如何获得第一部分:
indexes = np.random.randint(low=0, high=num_rows size=splitsize)
part1 = data[indexes, :]
如何获取数据的其他部分?
答案 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])