Previous topic

Setting up

Next topic

Installing NumPy

什么是NumPy?

NumPy是Python中用于科学计算的基础软件包。 它是一个Python库,提供多维数组对象,各种派生对象(如掩码数组和矩阵),以及用于数组快速操作的各种例程,包括数学,逻辑,形状操作,排序,选择,I / O离散傅立叶变换,基本线性代数,基本统计运算,随机模拟等等。

NumPy包的核心是ndarray对象。 这封装了同构数据类型的n维数组,其中许多操作在编译代码中执行以提高性能。 NumPy数组和标准Python序列之间有几个重要的区别:

  • NumPy数组在创建时具有固定大小,与Python列表(可以动态增长)不同。 更改ndarray的大小将创建一个新数组并删除原始数组。
  • NumPy数组中的元素都需要具有相同的数据类型,因此在内存中的大小相同。 例外:可以有(Python,包括NumPy)对象的数组,从而允许不同大小的元素的数组。
  • NumPy数组有助于对大量数据进行高级数学和其他类型的操作。 通常,与使用Python的内置序列相比,此类操作的执行效率更高,代码更少。
  • 越来越多的基于Python的科学和数学软件包正在使用NumPy数组;虽然这些通常支持Python序列输入,但它们在处理之前将这些输入转换为NumPy数组,并且它们通常输出NumPy数组。 换句话说,为了有效地使用当今大量(甚至大多数)基于Python的科学/数学软件,只知道如何使用Python的内置序列类型是不够的 - 还需要知道如何使用NumPy数组。

关于序列大小和速度的要点在科学计算中尤为重要。 举一个简单的例子,考虑将1-D序列中的每个元素与相同长度的另一个序列中的相应元素相乘的情况。 如果数据存储在两个Python列表中,ab,我们可以迭代每个元素:

c = []
for i in range(len(a)):
    c.append(a[i]*b[i])

这会产生正确答案,但如果ab每个包含数百万个数字,我们将为Python中循环效率低下付出代价。 我们可以通过写入在C中更快地完成相同的任务(为了清楚起见,我们忽略了变量声明和初始化,内存分配等)

for (i = 0; i < rows; i++): {
  c[i] = a[i]*b[i];
}

这节省了解释Python代码和操作Python对象所涉及的所有开销,但代价是从Python编码中获得的好处。 此外,所需的编码工作随着我们数据的维度而增加。 例如,在二维数组的情况下,C代码(如前所述)扩展为

for (i = 0; i < rows; i++): {
  for (j = 0; j < columns; j++): {
    c[i][j] = a[i][j]*b[i][j];
  }
}

NumPy为我们提供了两全其美的选择:当涉及ndarray时,逐个元素的操作是“默认模式”,但逐个元素的操作由预编译的C快速执行码。 In NumPy

c = a * b

以接近C的速度执行前面的示例所做的事情,但是我们期望基于Python的代码简化。 事实上,NumPy的成语更简单! 最后一个例子说明了NumPy的两个特征,它们是它的大部分功能的基础:矢量化和广播。

Vectorization描述了代码中没有任何显式的循环,索引等 - 这些事情当然只是在优化的,预编译的C代码中“幕后”。 矢量化代码有许多优点,其中包括:

  • 矢量化代码更简洁,更易于阅读
  • 更少的代码行通常意味着更少的错误
  • 代码更接近于标准的数学符号(通常,更容易正确编码数学结构)
  • 矢量化导致更多“Pythonic”代码。 如果没有矢量化,我们的代码就会被低效且难以阅读的for循环所困扰。

广播是用于描述操作的隐式逐元素行为的术语;一般而言,在NumPy中,所有操作,不仅仅是算术运算,而是逻辑,逐位,功能等,以这种隐式的逐元素方式表现,即它们广播。 此外,在上面的示例中,ab可以是相同形状的多维数组,或者是标量和数组,或者甚至是具有不同形状的两个数组,较小的阵列是“可扩展的”到更大的形状,使得得到的广播是明确的。 有关广播的详细“rules 规则”,请参阅numpy.doc.broadcasting

NumPy完全支持面向对象的方法,再次使用ndarray启动。 例如,ndarray是一个类,拥有许多方法和属性。 它的许多方法都在最外层的NumPy命名空间中镜像函数,使程序员可以完全自由地编写她喜欢的范例和/或最适合手头任务的范例。