获取所有可能的 numpy 数组列删除的列表

问题描述

给定以下 numpy 数组：

>>> a = np.arange(9).reshape((3,3))
>>> a
array([[0,1,2],[3,4,5],[6,7,8]])

如何获取所有可能的列删除列表？所以在这种情况下：

array([[[1,[4,[7,8]],[[0,1],4],7]]])

解决方法

您可以使用itertools.combinations：

>>> from itertools import combinations
>>> np.array([a[:,list(comb)] for comb in combinations(range(a.shape[1]),r=2)])

array([[[0,1],[3,4],[6,7]],[[0,2],5],8]],[[1,[4,[7,8]]])

,

或者，您可以先创建所需的列索引列表，然后使用 integer array indexing 从原始数组中选取所需的列：

r = range(a.shape[1])
cols = [[j for j in r if i != j] for i in r]  

cols
# [[1,[0,1]]

a[:,cols].swapaxes(0,1)

#[[[1 2]
#  [4 5]
#  [7 8]]
#
# [[0 2]
#  [3 5]
#  [6 8]]
#
# [[0 1]
#  [3 4]
#  [6 7]]]