【技术积累】Python中的NumPy库【一】

寻技术 Python编程 2023年07月12日 144

NumPy库是什么

NumPy是Python科学计算的核心库之一,用来进行科学计算,数值分析等矩阵运算。主要提供了以下几种功能:

1.多维数组(ndarray)对象,可以进行快速的数值计算和数组操作;

2.广播(Broadcast)功能,可以对不同形状的数组进行算术运算;

3.数学函数库,可以对数组进行各种数学运算;

4.线性代数、傅里叶变换、随机数生成等工具操作;

5.不同数据类型之间的传输功能;

6.数据的输入和输出功能。

在NumPy中,最重要的是它的ndarray对象,多为n维数组,因此它非常适合处理科学计算中的向量、矩阵等数据结构。numpy比列表更加高效,可以大大提高程序运行速度,是数据处理和科学计算的主要工具之一。

如何在NumPy中生成随机数?

在 NumPy 中,我们可以使用 random 模块来生成随机数。

下面是一些常用的随机数生成函数:

  1. np.random.rand():生成 0 到 1 之间的随机浮点数,可以接收任意个参数作为随机数生成器的形状。

  2. np.random.randn():生成符合标准正态分布的随机数,可以接收任意个参数作为随机数生成器的形状。

  3. np.random.randint():生成整数型的随机数,可以接收两个参数:最小值和最大值。

  4. np.random.random():生成 0 到 1 之间的随机浮点数,可以接收一个参数作为输出的形状。

  5. np.random.shuffle():随机打乱一个序列,可以接收一个参数作为要打乱的序列。

  6. np.random.choice():从指定的序列中随机挑选一个元素,可以接收两个参数:序列和挑选出元素的数量。

import numpy as np

# 生成一个形状为 (3, 2) 的随机浮点数数组
arr1 = np.random.rand(3, 2)

# 生成一个形状为 (3, 2) 的符合标准正态分布的随机数数组
arr2 = np.random.randn(3, 2)

# 生成 0 到 9 之间的随机整数
randint = np.random.randint(10)

# 生成一个形状为 (1, 3) 的随机浮点数数组
arr3 = np.random.random((1, 3))

# 打乱一个序列
arr4 = np.arange(10)
np.random.shuffle(arr4)

# 从指定的序列中随机挑选一个元素
choice = np.random.choice([1, 2, 3, 4, 5, 6], 1)

NumPy中NDArray对象有什么特点?

NumPy中NDArray(N-dimensional array)对象有以下特点:

  1. 它是多维数组,可以表示任意维度的数据。

  2. 所有元素必须是同一类型的数据。

  3. 它支持向量化操作,即对整个数组的操作只需要一条语句即可实现。

  4. 它快速且高效,因为它是使用C语言实现的,并且在内存使用和计算效率方面进行了优化。

  5. 它提供了大量的科学计算函数和方法,包括线性代数、傅里叶变换、统计分析等。

  6. 它易于与其他数据处理工具(如Pandas、SciPy等)集成使用,可以提高数据分析处理效率,并且支持大数据量的高效处理。

如何使用NumPy创建对角矩阵?

使用diag函数创建对角矩阵

import numpy as np
# 创建一个3x3的对角矩阵,对角线元素为1
diag_matrix = np.diag([1, 1, 1])
print(diag_matrix)
array([[1, 0, 0],
       [0, 1, 0],
       [0, 0, 1]])

你也可以在diag函数中使用一个数字来创建对应大小的对角矩阵。

# 创建一个5x5的对角矩阵,对角线元素为2
diag_matrix = np.diag(2*np.ones(5))
print(diag_matrix)
array([[2., 0., 0., 0., 0.],
       [0., 2., 0., 0., 0.],
       [0., 0., 2., 0., 0.],
       [0., 0., 0., 2., 0.],
       [0., 0., 0., 0., 2.]])

如何使用NumPy进行数组展平和拉伸操作?

NumPy提供了两个方法flatten()和ravel()都可用于将多维数组展平为一维数组,不同之处在于ravel()返回的是原数组的视图,而flatten()返回的是原数组的副本。下面是使用flatten()和ravel()方法展示多维数组的方法:

1.使用flatten()方法

import numpy as np

# 创建一个2x3的二维数组
a = np.array([[1, 2, 3], [4, 5, 6]])

# 展平数组
b = a.flatten()

print('原数组:')
print(a)

print('展平后的数组:')
print(b)
原数组:
[[1 2 3]
 [4 5 6]]

展平后的数组:
[1 2 3 4 5 6]

2.使用ravel()方法

import numpy as np

# 创建一个2x3的二维数组
a = np.array([[1, 2, 3], [4, 5, 6]])

# 拉伸数组
b = a.ravel()

print('原数组:')
print(a)

print('拉伸后的数组:')
print(b)
原数组:
[[1 2 3]
 [4 5 6]]

拉伸后的数组:
[1 2 3 4 5 6]

如何在NumPy中进行数组广播操作?

NumPy的数组广播操作可以在不同形状的数组之间进行数学运算,简化了数组的操作和计算。数组广播主要遵循以下规则:

  1. 数组维度不同,可以将维度较小的数组延伸为维度较大的数组,使它们的维度相等。
  2. 如果两个数组在某个维度上的形状相同,或者其中一个数组在该维度上的形状为1,则称它们在该维度上是兼容的,可以进行数学运算。
  3. 在任何维度上,如果一个数组形状为1,则可以沿着该维度扩展为另一个数组的对应维度。

以下是一个示例,演示如何使用numpy进行数组广播:

import numpy as np

#创建两个形状不同的数组
a = np.array([1, 2, 3])
b = np.array([[1], [2], [3]])

#打印两个数组的形状
print("a.shape = ", a.shape)
print("b.shape = ", b.shape)

#使用数组广播计算a和b的和
c = a + b
print("a + b = \n", c)
a.shape = (3,)
b.shape = (3, 1)
a + b =
 [[2 3 4]
 [3 4 5]
 [4 5 6]]

在这个示例中,我们创建了两个数组a和b,它们的形状不同。然后,我们使用数组广播计算a和b的和。由于a和b的形状不同,我们可以将维度较小的数组b延伸为维度相等的数组,以兼容它们之间的加法运算。最后,我们将结果存储在新数组c中。

如何使用NumPy进行数组的切片和子集选择?

NumPy可以使用切片和布尔索引来选择数组的子集。、

使用切片可以选择数组的某一部分,例如:

import numpy as np

a = np.array([1, 2, 3, 4, 5])
b = a[1:4]  # 选择a[1]到a[3]的元素
print(b)  # 输出 [2 3 4]

也可以对多维数组进行切片,例如:

a = np.array([[1, 2, 3],
              [4, 5, 6],
              [7, 8, 9]])
b = a[0:2, 1:3]  # 选择a中第1列到第2列、第2行到第3行的元素
print(b)  # 输出 [[2 3]
          #     [5 6]]

使用布尔索引可以根据某种条件选择数组中的元素,例如:

a = np.array([1, 2, 3, 4, 5])
b = a[a > 2]  # 选择a中大于2的元素
print(b)  # 输出 [3 4 5]

也可以对多维数组使用布尔索引,例如:

a = np.array([[1, 2, 3],
              [4, 5, 6],
              [7, 8, 9]])
b = a[a > 2]  # 选择a中大于2的元素
print(b)  # 输出 [3 4 5 6 7 8 9]

注意,布尔索引返回的是一维数组,需要使用reshape方法将其转换为多维数组。可以使用np.where来进行更复杂的条件选择。

如何使用NumPy进行矩阵运算和线性代数计算?

NumPy是Python语言的一个扩展库,专门处理大型多维数组与矩阵运算。NumPy提供了许多线性代数和矩阵操作函数。下面是使用NumPy进行矩阵运算的一些例子:

  • 创建矩阵

要使用NumPy创建一个矩阵,可以使用numpy.array()函数,并将二维数组传递给它。

import numpy as np

# 创建一个3x3的矩阵
matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print(matrix)

# 输出:
# [[1 2 3]
#  [4 5 6]
#  [7 8 9]]
  • 矩阵转置

要使用NumPy将矩阵转置,可以使用numpy.transpose()函数。

import numpy as np

# 创建一个3x3的矩阵
matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print(matrix)

# 转置矩阵
matrix_transpose = np.transpose(matrix)
print(matrix_transpose)

# 输出:
# [[1 2 3]
#  [4 5 6]
#  [7 8 9]]
# [[1 4 7]
#  [2 5 8]
#  [3 6 9]]
  • 矩阵相加和矩阵相减

要使用NumPy对矩阵进行加法和减法操作,可以使用numpy.add()和numpy.subtract()函数。

import numpy as np

# 创建两个矩阵
matrix_a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
matrix_b = np.array([[9, 8, 7], [6, 5, 4], [3, 2, 1]])

# 矩阵相加
matrix_sum = np.add(matrix_a, matrix_b)
print(matrix_sum)

# 矩阵相减
matrix_diff = np.subtract(matrix_a, matrix_b)
print(matrix_diff)

# 输出:
# [[10 10 10]
#  [10 10 10]
#  [10 10 10]]
# [[-8 -6 -4]
#  [-2  0  2]
#  [ 4  6  8]]
  • 矩阵乘法

要使用NumPy进行矩阵乘法,可以使用numpy.dot()函数。

import numpy as np

# 创建两个矩阵
matrix_a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
matrix_b = np.array([[9, 8, 7], [6, 5, 4], [3, 2, 1]])

# 矩阵乘法
matrix_prod = np.dot(matrix_a, matrix_b)
print(matrix_prod)

# 输出:
# [[ 30  24  18]
#  [ 84  69  54]
#  [138 114  90]]
  • 行列式和逆矩阵

要使用NumPy计算行列式和逆矩阵,可以使用numpy.linalg.det()和numpy.linalg.inv()函数。

import numpy as np

# 创建一个3x3的矩阵
matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 计算矩阵的行列式
matrix_det = np.linalg.det(matrix)
print(matrix_det)

# 计算矩阵的逆矩阵
matrix_inv = np.linalg.inv(matrix)
print(matrix_inv)

# 输出:
# 0.0
# [[-3.00000000e+00  6.00000000e+00 -3.00000000e+00]
#  [ 6.00000000e+00 -1.20000000e+01  6.00000000e+00]
#  [-3.00000000e+00  6.00000000e+00 -3.00000000e+00]]

这些只是NumPy中可用的许多矩阵运算之一。NumPy库还提供了其他矩阵运算,如解线性方程组等等 逆矩阵可以计算其它方面的线性代数问题,如解线性方程组,这是数据科学中不可或缺的。

如何使用NumPy进行数据的离散化和分组操作?

使用NumPy进行数据的离散化和分组操作可以使用numpy.digitize和numpy.bincount函数。

numpy.digitize函数可以将一组数据划分为不同的区间,返回每个数据所属的区间编号;numpy.bincount函数可以统计每个区间内数据的数量。这两个函数的联合使用可以实现数据的离散化和分组操作。

例如,对数据进行10个区间的离散化,并统计每个区间内数据的数量可以使用以下代码:

import numpy as np

data = np.array([1, 4, 8, 3, 6, 9, 2, 5, 7, 10])
bins = np.linspace(0, 10, 11) # 划分为10个区间
indices = np.digitize(data, bins) # 划分数据所属区间
counts = np.bincount(indices)

上述代码执行完毕后,counts数组中即为每个区间内数据的数量。可以在此基础上进行更多操作,例如计算每个区间内数据的平均值等。

NumPy如何进行数组的输入和输出?

 

NumPy可以通过多种方式进行数组的输入和输出,包括文本文件、二进制文件、内存缓冲等。

下面是一些常见的数组输入和输出的方法和示例:

  1. 从文本文件读取数组:
  2. 将数组保存到文本文件中:
  3. 从二进制文件读取数组:
  4. 将数组保存到二进制文件中:
  5. 从内存缓冲读取数组:
  6. 将数组保存到内存缓冲中:
import numpy as np
import io

# 从文本文件读取数组
data = np.loadtxt('data.txt')
print(data)

# 将数组保存到文本文件中
data = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
np.savetxt('data.txt', data)

# 从二进制文件读取数组
data = np.load('data.npy')
print(data)

# 将数组保存到二进制文件中
data = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
np.save('data.npy', data)

# 从内存缓冲读取数组
str_data = '1 2 3\n4 5 6\n7 8 9'
data = np.genfromtxt(str_data.splitlines(), delimiter=' ')
print(data)

# 将数组保存到内存缓冲中
data = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
output = io.StringIO()
np.savetxt(output, data, delimiter=',')
str_data = output.getvalue()
print(str_data)

除此之外,NumPy还提供了很多其他的输入和输出方式,如Pandas DataFrame、HDF5等。具体可以查阅官方文档。

关闭

用微信“扫一扫”