1、数组或矩阵中替换大于或小于等于某个值的元素
最快和最简洁的方法是使用Numpy的内置索引。
import numpy as np
a = np.array([[1,2,3],
[1,2,3],
[1,2,3]])
a[a > 2] = 0
print(a)
结果:
[[1 2 0]
[1 2 0]
[1 2 0]]
2、把二维数组(矩阵)中行或列的值相加
1)、全部相加
import numpy as np
a = np.array([[1,2,3],
[1,2,3],
[1,2,3]])
print(a.sum())
结果
18
2)、行或列相加
print("行相加:")
print(a.sum(axis=1))
print("列相加:")
print(a.sum(axis=0))
结果
行相加:
[6 6 6]
列相加:
[3 6 9]
3)、指定行或列相加
print("倒数第一行相加:")
print(a[-1].sum())
print("倒数第一列相加:")
print(a[:, -1].sum())
结果
倒数第一行相加:
6
倒数第一列相加:
9