数学建模Python numpy常用操作总结

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