numpy快速入门(2):基本操作

2018-09-02  本文已影响0人  大锅烩菜

1. 判断元素是否等于某个值

matrix = numpy.array([[3,6,9],[2,4,8],[1,5,7]])

# == 对矩阵中每一个元素进行判断
res = matrix == 4
print(res)

结果:

[[False False False]
 [False  True False]
 [False False False]]

2. 将所有值为10的元素变为20

首先,判断每一个元素是否是10,之后将这个结果作为索引,使满足条件为True的所有元素的值变为20

matrix = numpy.array([[3,10,5],[4,5,10],[1,9,10]])

#1. 判断元素是否是10
res = matrix ==10
# 将值为10的元素修改为20
matrix[res] = 20
print(matrix)

结果:

[[ 3 20  5]
 [ 4  5 20]
 [ 1  9 20]]

3. 或运算

或用于多条件判断。

vector = numpy.array([9,10,5])
# 判断元素是否等于10或等于5
res = (vector==10)|(vector==5)
print(res)   
-------------
[False  True  True]

4. 类型转换

vector =numpy.array(["1","2","3"])
print(vector.dtype)
print(vector)
vector = vector.astype(float)
print(vector.dtype)
print(vector)

结果:

<U1
['1' '2' '3']
float64
[1. 2. 3.]

5. 最小值

matrix = numpy.array([[3,10,5],[4,5,10],[1,9,10]])
print(matrix.min())

最大值是max函数

6. 求和

matrix = numpy.array([[3,10,5],[4,5,10],[1,9,10]])
print(matrix.sum()) # 57

按照行求和:

matrix = numpy.array([[3,10,5],[4,5,10],[1,9,10]])
print(matrix.sum(axis=1)) # [18 19 20]

按照列求和:

matrix = numpy.array([[3,10,5],[4,5,10],[1,9,10]])
print(matrix.sum(axis=0)) # [ 8 24 25]
上一篇 下一篇

猜你喜欢

热点阅读