Question

我有一个类似的方阵

0 0 0 0 0 0 0
1 0 0 0 1 0 0
1 1 1 0 0 0 0
0 0 0 0 0 0 0
1 1 1 0 1 1 0
1 1 1 0 1 1 0
0 0 0 0 0 0 0

例如，此输出为：

0 0 0 | 0 0 |
1 0 0 | 1 0 |
1 1 1 | 0 0 |
- - - + - - +
1 1 1 | 1 1 |
1 1 1 | 1 1 |
- - - + - - +

0 0 0 0 0
1 0 0 1 0
1 1 1 0 0
1 1 1 1 1
1 1 1 1 1

注意第4行和第0列以及最后一个都为0。我仅当第i行和第i列全为0时，才删除行和列。（还请注意，由于第一列包含非零元素，因此保留了第一行0。）

是否有一种干净而简单的方法来做到这一点而不遍历每个人？

Answer 1

假设a是一个在两个维度上具有相同大小的numpy数组：

# find out the index to keep
keep_idx = a.any(0) | a.any(1)

# subset the array
a[keep_idx][:, keep_idx]
#array([[0, 0, 0, 0, 0],
#       [1, 0, 0, 1, 0],
#       [1, 1, 1, 0, 0],
#       [1, 1, 1, 1, 1],
#       [1, 1, 1, 1, 1]])

Answer 2

假设我们有一个7 * 7数据框，类似于您的矩阵，那么下面的代码就可以了：

row_sum = df.sum(axis=1)
col_sum = df.sum(axis=0)

lst=[]

for i in range(len(df)):
    if ((row_sum[i] == 0) & (col_sum[i]==0)):
        lst.append(i)

df1 = df.drop(lst, axis = 1).drop(lst, axis = 0)

舍弃全部为0 {

2 个答案: