numpy

numpy

import numpy as np #导入numpy

x = np.array([1.0, 2.0 ,3.0]) #生成numpy数组
print(x)

[1. 2. 3.]

A = np.array([[1, 2], [3, 4]])
print(A)
A.shape           #使用shape查看矩阵的形状
A.dtype           #使用dtype查看矩阵元素的数据类型

[[1 2]
 [3 4]]





dtype('int32')

B = np.array([[3, 0],[0,6]])
A + B
A * B

array([[ 3,  0],
       [ 0, 24]])

print(A)
A * 10

[[1 2]
 [3 4]]





array([[10, 20],
       [30, 40]])

广播

A = np.array([[1, 2],[3, 4]])
B = np.array([10,20])
A * B

array([[10, 40],
       [30, 80]])

访问元素

import numpy as np

x = np.array([[51, 55], [14, 19], [0, 4]])
print(x)

[[51 55]
 [14 19]
 [ 0  4]]

x[0]

array([51, 55])

x[0][1]

55

for row in x:
    print(row)

[51 55]
[14 19]
[0 4]

x = x.flatten()#将x转换为一维数组
print(x)

[51 55 14 19  0  4]

x[np.array([0, 2, 4])]#获取索引0，2，4的元素s

array([51, 14,  0])