numpy基本的使用方法

一，导入numpy模块

import numpy as np

1

二，创建一个2行3列的矩阵

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

输出维度

print(‘number of dim:’, array.ndim)

输出行数和列数

print(‘shape :’, array.shape)

输出元素个数

print(‘size:’, array.size)

1

2

3

4

5

6

7

8

运行显示

三，创建一个三维数组

arr = np.array([[[1,2],[3,4],[5,6]],[[7,8],[9,10],[10,11]]])

print(arr)

print(arr.ndim)

print(arr.dtype)

print(arr.shape)

print(arr.size)

1

2

3

4

5

6

运行显示

四，创建一个2行5列的矩阵，大小10以内

arr = np.arange(10).reshape(2,5)

print(arr)

1

2

五，创建一个三维数组

arr = np.arange(27).reshape(3,3,3)

print(arr)

1

2

六，随机创建3行2列的矩阵（重点使用）

arr = np.random.rand(3,2)

print(arr)

1

2

七，创建2行3列的0矩阵

arr =np.zeros((2,3))

print(arr)

1

2

八，创建2行3列的1矩阵

arr = np.ones((2,3))

print(arr)

1

2

九，找出矩阵中大于5的数字

arr = np.arange(10).reshape(2, 5)

print(arr)

print(np.where(arr > 5), arr, 0)

1

2

3

十，创建一个4行5列矩阵 数值1-10， 一共20个数字。 取某一行，某一列

arr = np.random.randint(1, 10, 20).reshape(4, 5)

创建一个4行5列矩阵 数值1-10 一共20个数字

print(arr)

print(arr[0]) # 取第一行数组

print(arr[0, 0]) # 取第1行第1列数字

print(arr[:, 0]) # 取第一列数组 （常用）

1

2

3

4

5

6

arr = np.random.randint(1, 10, 20).reshape(4, 5)

print(arr)

print(arr[0:4:3]) # 取第1行和第4行

print(arr[[0, 3]]) # 取第1行和第4行(常用)

1

2

3

4

arr = np.random.randint(1, 10, 20).reshape(4, 5)

print(arr)

print(arr[0:, 0:5:4]) # 第1列和第4列

print(arr[0:, [0, 4]]) # 第1列和第4列（常用）

1

2

3

4

arr = np.random.randint(1, 10, 20).reshape(4, 5)

print(arr)

print(arr[[1, 2]]) # 取第2行和第3行

print(arr[[1, 2], 1:3]) # 取第2，3行和 2，3列

1

2

3

4

十一，遍历矩阵

arr = np.random.randint(1, 10, 20).reshape(4, 5)

print(arr)

print(arr[0][0]) # 取第一行，第一列的数字

print(arr.shape[0])

for i in range(0,arr.shape[0]):

for j in range(arr.shape[1]):

   print(arr[i][j], end=" ")

1

2

3

4

5

6

7