数据:Numpy去掉全零行或列np.all()

1. np.all()作用

1. np.all(arr,axis=0)，表示在垂直方向上计算元素的与运算（AND）
2. np.all(arr,axis=1)，表示在水平方向上计算元素的与运算（AND）

arr = np.array([[1, 1], [0, 1]])

boolean = np.all(arr, axis=1)  # 矩阵行方向做与运算
boolean2 = np.all(arr, axis=0)  # 矩阵列方向做与运算
print(boolean)

print("*" * 20)
print(boolean2)

输出：
[ True False]
********************
[False  True]

2. 去除全零行或列

arr = np.array([[0, 1, 2], [0, 2, 3], [0, 0, 0]])

print(arr)

# 去除全零行
not_row = arr[[not np.all(arr[i] == 0) for i in range(arr.shape[0])], :]
print("*" * 20)
print(not_row)

# 去除全零列
bot_col = arr[:, [not np.all(arr[:, i] == 0) for i in range(arr.shape[1])]]
print("*" * 20)
print(bot_col)

输出：
[[0 1 2]
 [0 2 3]
 [0 0 0]]
********************
[[0 1 2]
 [0 2 3]]
********************
[[1 2]
 [2 3]
 [0 0]]

这篇好文章是转载于：学新通技术网