NumPy教程

作者|ANIRUDDHA BHANDARI
编译|VK
来源|Analytics Vidhya

概述

• NumPy是一个Python库，每个数据科学专业人员都应该熟悉它
• 这个全面的NumPy教程从头开始介绍NumPy，从基本的数学运算到NumPy如何处理图像数据
• 本文中有大量的Numpy概念和Python代码

介绍

我非常喜欢Python中的NumPy库。在我的数据科学之旅中，我无数次依赖它来完成各种任务，从基本的数学运算到使用它进行图像分类！

简而言之，NumPy是Python中最基本的库之一，也许是其中最有用的库。NumPy高效地处理大型数据集。 作为一名数据科学家或一名有抱负的数据科学专业人士，我们需要对NumPy及其在Python中的工作原理有一个扎实的掌握。

在本文中，我将首先描述一下NumPy库是什么，以及为什么你应该选择它而不是繁琐的Python列表。然后，我们将介绍一些最基本的NumPy操作，这些操作将使你喜欢这个很棒的库！

目录

• NumPy库是什么？
• Python列表与NumPy数组有什么区别？

• 创建NumPy数组

  • 基本的ndarray
  • 全零数组
  • 全一数组
  • ndarray中的随机数
  • 定制的数组
  • NumPy的Imatrix
  • 等间距的ndarray

• NumPy数组的形状与重塑

  • NumPy数组的维数
  • NumPy数组的形状
  • NumPy数组的大小
  • 重塑NumPy数组
  • 展开NumPy数组
  • NumPy数组的转置

• 扩展和压缩一个NumPy数组

  • 展开NumPy数组
  • 压缩NumPy数组

• NumPy数组的索引与切片

  • 一维数组的切片
  • 二维数组切片
  • 三维数组切片
  • NumPy数组的负切片

• 堆叠和级联Numpy数组

  • 堆叠ndarrays
  • 级联ndarrays
• Numpy数组广播
• NumPy Ufuncs

• 用NumPy数组计算

  • 平均值、中位数和标准差
  • 最小最大值及其索引
• 在NumPy数组中排序
• NumPy数组和图像

NumPy库是什么？

NumPy是Python数值库，是Python编程中最有用的科学库之一。它支持大型多维数组对象和各种工具。各种其他的库，如Pandas、Matplotlib和Scikit-learn，都建立在这个令人惊叹的库之上。

数组是元素/值的集合，可以有一个或多个维度。一维数组称为向量，二维数组称为矩阵。

NumPy数组称为ndarray或N维数组，它们存储相同类型和大小的元素。它以其高性能而闻名，并在数组规模不断扩大时提供高效的存储和数据操作。

下载Anaconda时，NumPy会预先安装。但是如果你想在你的机器上单独安装NumPy，只需在你的终端上键入以下命令：

pip install numpy

现在需要导入库：

import numpy as np

np实际上是数据科学界使用的NumPy的缩写。

Python列表与NumPy数组有什么区别？

如果你熟悉Python，你可能会想，既然我们已经有了Python列表，为什么还要使用NumPy数组？毕竟，这些Python列表充当一个数组，可以存储各种类型的元素。这是一个完全正确的问题，答案隐藏在Python在内存中存储对象的方式中。

Python对象实际上是一个指向内存位置的指针，该内存位置存储有关该对象的所有详细信息，如字节和值。尽管这些额外的信息使Python成为一种动态类型语言，但它也付出了代价，这在存储大量对象（如在数组中）时变得显而易见。

Python列表本质上是一个指针数组，每个指针指向一个包含与元素相关信息的位置。这在内存和计算方面增加了很多开销。当列表中存储的所有对象都是同一类型时，大多数信息都是冗余的！

为了解决这个问题，我们使用只包含同构元素的NumPy数组，即具有相同数据类型的元素。这使得它在存储和操作数组方面更加高效。

当数组包含大量元素（比如数千或数百万个元素）时，这种差异就变得明显了。另外，使用NumPy数组，你可以执行元素操作，这是使用Python列表不可能做到的！

这就是为什么在对大量数据执行数学操作时，NumPy数组比Python列表更受欢迎的原因。

创建NumPy数组

基本的ndarray

考虑到NumPy数组解决的复杂问题，它很容易创建。要创建一个非常基本的ndarray，可以使用np.array()方法。你只需将数组的值作为列表传递：

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

输出：

array([1, 2, 3, 4])

此数组包含整数值。可以在dtype参数中指定数据类型：

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

输出：

array([1., 2., 3., 4.], dtype=float32)

由于NumPy数组只能包含同构数据类型，因此如果类型不匹配，则将向上转换值：

np.array([1,2.0,3,4])

输出：

array([1., 2., 3., 4.])

在这里，NumPy将整数值上移到浮点值。

NumPy数组也可以是多维的。

np.array([[1,2,3,4],[5,6,7,8]])

array([[1, 2, 3, 4],
       [5, 6, 7, 8]])

在这里，我们创建了一个二维数组。

注：矩阵只是一个NxM形状的数字矩形数组，其中N是行数，M是矩阵中的列数。你刚才看到的是一个2x4矩阵。

全零数组

NumPy允许你使用 np.zeros()方法。你只需传递所需数组的形状：

np.zeros(5)

array([0., 0., 0., 0., 0.])

上面一个是一维数组，下面一个是二维数组：

np.zeros((2,3))

array([[0., 0., 0.],
       [0., 0., 0.]])

全一数组

你还可以使用 np.ones()方法获得全一数组：

np.ones(5,dtype=np.int32)

array([1, 1, 1, 1, 1])

ndarray中的随机数

创建ndarray的另一个非常常用的方法是随机随机数方法。它创建一个给定形状的数组，其随机值来自[0,1]：

# 随机的
np.random.rand(2,3)

array([[0.95580785, 0.98378873, 0.65133872],
       [0.38330437, 0.16033608, 0.13826526]])

定制的数组

或者，实际上，可以使用 np.full()方法。只需传入所需数组的形状和所需的值：

np.full((2,2),7)

array([[7, 7],
       [7, 7]])

NumPy的Imatrix

另一个伟大的方法是np.eye()返回一个数组，其对角线上有1，其他地方都有0。

一个单位矩阵是一个正方形矩阵，它的主对角线上有1，其他地方都有0。下面是形状为3x 3的单位矩阵。

注：正方形矩阵是N x N的形状。这意味着它具有相同数量的行和列。

# 单位矩阵
np.eye(3)

array([[1., 0., 0.],
       [0., 1., 0.],
       [0., 0., 1.]])

但是，NumPy允许你灵活地更改对角线，值必须为1。你可以将其移到主对角线上方：

# 不是单位矩阵

np.eye(3,k=1)

array([[0., 1., 0.],
       [0., 0., 1.],
       [0., 0., 0.]])

或将其移到主对角线下方：

np.eye(3,k=-2)

array([[0., 0., 0.],
       [0., 0., 0.],
       [1., 0., 0.]])

注：只有当1沿主对角线而不是任何其他对角线时，矩阵才称为单位矩阵！

等间距的ndarray

你可以使用np.arange()方法：

np.arange(5)

array([0, 1, 2, 3, 4])

通过分别传递三个数字作为这些值的参数，可以显式定义值间隔的开始、结束和步长。这里要注意的一点是，间隔定义为[开始，结束)，其中最后一个数字将不包含在数组中：

np.arange(2,10,2)

array([2, 4, 6, 8])

由于步长定义为2，因此下图展示了输出的元素。注意，10不会打印出来，因为它是最后一个元素。

另一个类似的功能是 np.linspace()，但它将从间隔中获取需要检索的样本数，而不是步长。这里要注意的一点是，最后一个数字包含在返回的值中，这与np.arange()不同

np.linspace(0,1,5)

array([0.  , 0.25, 0.5 , 0.75, 1.  ])

太好了！现在你知道了如何使用NumPy创建数组。但了解数组的形状也很重要。

NumPy数组的形状与重塑

创建了ndarray之后，接下来要做的是检查ndarray的轴数、形状和大小。

NumPy数组的维数

可以使用ndims属性轻松确定NumPy数组的维数或轴数：

# 轴数
a = np.array([[5,10,15],[20,25,20]])
print('Array :','\n',a)
print('Dimensions :','\n',a.ndim)

Array : 
 [[ 5 10 15]
 [20 25 20]]
Dimensions : 
 2

这个数组有两个维度：2行3列。

NumPy数组的形状

形状是NumPy数组的一个属性，它显示每个维度上有多少行元素。你可以进一步索引ndarray返回的形状，以便沿每个维度获取值：

a = np.array([[1,2,3],[4,5,6]])
print('Array :','\n',a)
print('Shape :','\n',a.shape)
print('Rows = ',a.shape[0])
print('Columns = ',a.shape[1])

Array : 
 [[1 2 3]
 [4 5 6]]
Shape : 
 (2, 3)
Rows =  2
Columns =  3

NumPy数组的大小

可以使用size属性确定数组中有多少值。它只是将行数乘以ndarray中的列数：

# size of array
a = np.array([[5,10,15],[20,25,20]])
print('Size of array :',a.size)
print('Manual determination of size of array :',a.shape[0]*a.shape[1])

Size of array : 6
Manual determination of size of array : 6

重塑NumPy数组

可以使用np.reshape()方法。它在不更改ndarray中的数据的情况下更改ndarray的形状：

# 重塑
a = np.array([3,6,9,12])
np.reshape(a,(2,2))

array([[ 3,  6],
       [ 9, 12]])

在这里，我将ndarray从一维重塑为二维ndarray。

重塑时，如果你不确定任何轴的形状，只需输入-1。当NumPy看到-1时，它会自动计算形状：

a = np.array([3,6,9,12,18,24])
print('Three rows :','\n',np.reshape(a,(3,-1)))
print('Three columns :','\n',np.reshape(a,(-1,3)))

Three rows : 
 [[ 3  6]
 [ 9 12]
 [18 24]]
Three columns : 
 [[ 3  6  9]
 [12 18 24]]

展开NumPy数组

有时，当你有多维数组并希望将其折叠为一维数组时，可以使用 flatten()方法或ravel()方法：

a = np.ones((2,2))
b = a.flatten()
c = a.ravel()
print('Original shape :', a.shape)
print('Array :','\n', a)
print('Shape after flatten :',b.shape)
print('Array :','\n', b)
print('Shape after ravel :',c.shape)
print('Array :','\n', c)

Original shape : (2, 2)
Array : 
 [[1. 1.]
 [1. 1.]]
Shape after flatten : (4,)
Array : 
 [1. 1. 1. 1.]
Shape after ravel : (4,)
Array : 
 [1. 1. 1. 1.]

但是flatten() 和ravel()之间的一个重要区别是前者返回原始数组的副本，而后者返回对原始数组的引用。这意味着对ravel()返回的数组所做的任何更改也将反映在原始数组中，而flatten()则不会这样。

b[0] = 0
print(a)

[[1. 1.]
 [1. 1.]]

所做的更改没有反映在原始数组中。

c[0] = 0
print(a)

[[0. 1.]
 [1. 1.]]

但在这里，更改后的值也反映在原始ndarray中。

这里发生的事情是flatten()创建了ndarray的深层副本，而ravel()创建了ndarray的浅层副本。

深层副本意味着在内存中创建了一个全新的ndarray，flatten()返回的ndarray对象现在指向这个内存位置。因此，此处所做的任何更改都不会反映在原始ndarray中。

另一方面，浅拷贝返回对原始内存位置的引用。这意味着ravel()返回的对象指向与原始ndarray对象相同的内存位置。因此，毫无疑问，对该ndarray所做的任何更改也将反映在原始ndarray中。

NumPy数组的转置

NumPy的另一个非常有趣的重塑方法是transpose()方法。它接受输入数组并用列值交换行，用行值交换列值：

a = np.array([[1,2,3],
[4,5,6]])
b = np.transpose(a)
print('Original','\n','Shape',a.shape,'\n',a)
print('Expand along columns:','\n','Shape',b.shape,'\n',b)

Original 
 Shape (2, 3) 
 [[1 2 3]
 [4 5 6]]
Expand along columns: 
 Shape (3, 2) 
 [[1 4]
 [2 5]
 [3 6]]

在转置一个2x 3数组时，我们得到了一个3x2数组。转置在线性代数中有着重要的意义。

扩展和压缩一个NumPy数组

展开NumPy数组

通过提供要展开的数组和轴，可以使用expand_dims()方法将新轴添加到数组中：

# 展开维度
a = np.array([1,2,3])
b = np.expand_dims(a,axis=0)
c = np.expand_dims(a,axis=1)
print('Original:','\n','Shape',a.shape,'\n',a)
print('Expand along columns:','\n','Shape',b.shape,'\n',b)
print('Expand along rows:','\n','Shape',c.shape,'\n',c)

Original: 
 Shape (3,) 
 [1 2 3]
Expand along columns: 
 Shape (1, 3) 
 [[1 2 3]]
Expand along rows: 
 Shape (3, 1) 
 [[1]
 [2]
 [3]]

压缩NumPy数组

另一方面，如果希望减小数组的轴，请使用squeeze()方法。它将删除具有单个条目的轴。这意味着，如果创建了一个2 x 2 x 1矩阵，则squeze()将从矩阵中删除第三个维度：

# squeeze
a = np.array([[[1,2,3],
[4,5,6]]])
b = np.squeeze(a, axis=0)
print('Original','\n','Shape',a.shape,'\n',a)
print('Squeeze array:','\n','Shape',b.shape,'\n',b)

Original 
 Shape (1, 2, 3) 
 [[[1 2 3]
  [4 5 6]]]
Squeeze array: 
 Shape (2, 3) 
 [[1 2 3]
 [4 5 6]]

但是，如果你已经有一个2×2的矩阵，在这种情况下使用squeeze()会给你一个错误:

# squeeze
a = np.array([[1,2,3],
[4,5,6]])
b = np.squeeze(a, axis=0)
print('Original','\n','Shape',a.shape,'\n',a)
print('Squeeze array:','\n','Shape',b.shape,'\n',b)

NumPy数组的索引与切片

到目前为止，我们已经看到了如何创建一个NumPy数组以及如何处理它的形状。在本节中，我们将看到如何使用索引和切片从数组中提取特定值。

一维数组的切片

切片意味着从一个索引检索元素到另一个索引。我们要做的就是像这样[start: end]

然而，你也可以定义步长。例如你可以将步长定义为2，这意味着让元素远离当前索引2个位置进行取值。

将所有这些内容合并到一个索引中看起来像这样： [start​：end：step-size]。

a = np.array([1,2,3,4,5,6])
print(a[1:5:2])

[2 4]

注意，没有考虑最后一个元素。这是因为切片包括开始索引，但不包括结束索引。

解决方法是将下一个更高的索引写入要检索的最终索引值：

a = np.array([1,2,3,4,5,6])
print(a[1:6:2])

[2 4 6]

如果不指定起始索引或结束索引，则默认值分别为0或数组大小。默认情况下步长为1。

a = np.array([1,2,3,4,5,6])
print(a[:6:2])
print(a[1::2])
print(a[1:6:])

[1 3 5]
[2 4 6]
[2 3 4 5 6]

二维数组切片

现在，二维数组有行和列，所以对二维数组进行切片会有点困难。但是一旦你理解了它，你就可以分割任何维度数组！

在学习如何分割二维数组之前，让我们先看看如何从二维数组中检索元素：

a = np.array([[1,2,3],
[4,5,6]])
print(a[0,0])
print(a[1,2])
print(a[1,0])

1
6
4

在这里，我们提供了行值和列值来标识要提取的元素。在一维数组中，我们只提供列值，因为只有一行。

因此，要对二维数组进行切片，需要同时提到行和列的切片：

a = np.array([[1,2,3],[4,5,6]])
# 打印第一行值
print('First row values :','\n',a[0:1,:])
# 具有列的步长
print('Alternate values from first row:','\n',a[0:1,::2])
# 
print('Second column values :','\n',a[:,1::2])
print('Arbitrary values :','\n',a[0:1,1:3])

First row values : 
 [[1 2 3]]
Alternate values from first row: 
 [[1 3]]
Second column values : 
 [[2]
 [5]]
Arbitrary values : 
 [[2 3]]

三维数组切片

到目前为止我们还没有看到三维数组。首先让我们想象一下三维数组的样子：

a = np.array([[[1,2],[3,4],[5,6]],# 第一个轴数组
[[7,8],[9,10],[11,12]],# 第二个轴数组
[[13,14],[15,16],[17,18]]])# 第三个轴数组
# 3-D array
print(a)

[[[ 1  2]
  [ 3  4]
  [ 5  6]]

 [[ 7  8]
  [ 9 10]
  [11 12]]

 [[13 14]
  [15 16]
  [17 18]]]

除了行和列之外，在二维数组中，三维数组还有一个深度轴，在这个深度轴上，它将一个二维数组放在另一个数组后面。所以，当你在切片一个三维数组时，你还需要提到你要切片哪个二维数组。这通常作为索引中的第一个值出现：

# value
print('First array, first row, first column value :','\n',a[0,0,0])
print('First array last column :','\n',a[0,:,1])
print('First two rows for second and third arrays :','\n',a[1:,0:2,0:2])

First array, first row, first column value : 
 1
First array last column : 
 [2 4 6]
First two rows for second and third arrays : 
 [[[ 7  8]
  [ 9 10]]

 [[13 14]
  [15 16]]]

如果希望将值作为一维数组，则可以始终使用flatten()方法来完成此项工作！

print('Printing as a single array :','\n',a[1:,0:2,0:2].flatten())

Printing as a single array : 
 [ 7  8  9 10 13 14 15 16]

NumPy数组的负切片

对数组进行切片的一个有趣的方法是使用负切片。负切片从末尾而不是开头打印元素。请看下面：

a = np.array([[1,2,3,4,5],
[6,7,8,9,10]])
print(a[:,-1])

[ 5 10]

这里，打印了每行的最后一个值。但是，如果我们想从末尾提取，我们必须显式地提供负步长，否则结果将是空列表。

print(a[:,-1:-3:-1])

[[ 5  4]
 [10  9]]

尽管如此，切片的基本逻辑保持不变，即输出中从不包含结束索引。

负切片的一个有趣的用途是反转原始数组。

a = np.array([[1,2,3,4,5],
[6,7,8,9,10]])
print('Original array :','\n',a)
print('Reversed array :','\n',a[::-1,::-1])

Original array : 
 [[ 1  2  3  4  5]
 [ 6  7  8  9 10]]
Reversed array : 
 [[10  9  8  7  6]
 [ 5  4  3  2  1]]

也可以使用flip()方法来反转ndarray。

a = np.array([[1,2,3,4,5],
[6,7,8,9,10]])
print('Original array :','\n',a)
print('Reversed array vertically :','\n',np.flip(a,axis=1))
print('Reversed array horizontally :','\n',np.flip(a,axis=0))

Original array : 
 [[ 1  2  3  4  5]
 [ 6  7  8  9 10]]
Reversed array vertically : 
 [[ 5  4  3  2  1]
 [10  9  8  7  6]]
Reversed array horizontally : 
 [[ 6  7  8  9 10]
 [ 1  2  3  4  5]]

堆叠和级联Numpy数组

堆叠ndarrays

可以通过组合现有数组来创建新数组。你可以通过两种方式来完成：

• 使用vstack()方法垂直组合数组（即沿行），从而增加结果数组中的行数
• 或者使用hstack()以水平方式（即沿列）组合数组，从而增加结果数组中的列数

a = np.arange(0,5)
b = np.arange(5,10)
print('Array 1 :','\n',a)
print('Array 2 :','\n',b)
print('Vertical stacking :','\n',np.vstack((a,b)))
print('Horizontal stacking :','\n',np.hstack((a,b)))

Array 1 : 
 [0 1 2 3 4]
Array 2 : 
 [5 6 7 8 9]
Vertical stacking : 
 [[0 1 2 3 4]
 [5 6 7 8 9]]
Horizontal stacking : 
 [0 1 2 3 4 5 6 7 8 9]

这里要注意的一点是，组合数组的轴应该具有相同的大小，否则一定会出错！

a = np.arange(0,5)
b = np.arange(5,9)
print('Array 1 :','\n',a)
print('Array 2 :','\n',b)
print('Vertical stacking :','\n',np.vstack((a,b)))
print('Horizontal stacking :','\n',np.hstack((a,b)))

组合数组的另一个有趣的方法是使用dstack()方法。它按索引组合数组元素，并沿深度轴堆叠它们：

a = [[1,2],[3,4]]
b = [[5,6],[7,8]]
c = np.dstack((a,b))
print('Array 1 :','\n',a)
print('Array 2 :','\n',b)
print('Dstack :','\n',c)
print(c.shape)

Array 1 : 
 [[1, 2], [3, 4]]
Array 2 : 
 [[5, 6], [7, 8]]
Dstack : 
 [[[1 5]
  [2 6]]

 [[3 7]
  [4 8]]]
(2, 2, 2)

级联ndarrays

虽然堆叠数组是组合旧数组以获得新数组的一种方法，但也可以使用concatenate()方法，其中传递的数组沿现有轴连接：

a = np.arange(0,5).reshape(1,5)
b = np.arange(5,10).reshape(1,5)
print('Array 1 :','\n',a)
print('Array 2 :','\n',b)
print('Concatenate along rows :','\n',np.concatenate((a,b),axis=0))
print('Concatenate along columns :','\n',np.concatenate((a,b),axis=1))

Array 1 : 
 [[0 1 2 3 4]]
Array 2 : 
 [[5 6 7 8 9]]
Concatenate along rows : 
 [[0 1 2 3 4]
 [5 6 7 8 9]]
Concatenate along columns : 
 [[0 1 2 3 4 5 6 7 8 9]]

此方法的缺点是原始数组必须具有要合并的轴。否则，准备好迎接错误。

另一个非常有用的函数是append方法，它将新元素添加到ndarray的末尾。当你已经有了一个现有的ndarray，但希望向其添加新值时，这显然很有用。

# 将值附加到ndarray
a = np.array([[1,2],
             [3,4]])
np.append(a,[[5,6]], axis=0)

array([[1, 2],
       [3, 4],
       [5, 6]])

Numpy数组广播

广播是ndarrays最好的功能之一。它允许你在不同大小的ndarray之间或ndarray与简单数字之间执行算术运算！

广播基本上延伸较小的ndarray，使其与较大ndarray的形状相匹配：

a = np.arange(10,20,2)
b = np.array([[2],[2]])
print('Adding two different size arrays :','\n',a+b)
print('Multiplying an ndarray and a number :',a*2)

Adding two different size arrays : 
 [[12 14 16 18 20]
 [12 14 16 18 20]]
Multiplying an ndarray and a number : [20 24 28 32 36]

它的工作可以看作是拉伸或复制标量，即数字，[2，2，2]以匹配ndarray的形状，然后按元素执行操作。但是实际上并没有生成该数组，这只是一种思考广播如何运作的方式。

这非常有用，因为用标量值乘一个数组比用另一个数组更有效！需要注意的是，只有当两个ndarray兼容时，它们才能一起广播。

Ndarrays在以下情况下兼容：

1. 两者都有相同的尺寸
2. 其中一个ndarrays的维数是1。尺寸为1的广播会满足尺寸更大的ndarray的大小要求

如果数组不兼容，你将得到一个ValueError。

a = np.ones((3,3))
b = np.array([2])
a+b

array([[3., 3., 3.],
       [3., 3., 3.],
       [3., 3., 3.]])

在这里，假设第二个ndarray被拉伸成3x 3形状，然后计算结果。

NumPy Ufuncs

Python是一种动态类型语言。这意味着在赋值时不需要知道变量的数据类型。Python将在运行时自动确定它。虽然这意味着编写更干净、更容易的代码，但也会使Python变得迟缓。

当Python必须重复执行许多操作（比如添加两个数组）时，这个问题就会显现出来。这是因为每次需要执行操作时，Python都必须检查元素的数据类型。使用ufuncs函数的NumPy可以解决这个问题。

NumPy使这个工作更快的方法是使用向量化。向量化在编译的代码中以逐元素的方式对ndarray执行相同的操作。因此，不需要每次都确定元素的数据类型，从而执行更快的操作。

ufuncs 是NumPy中的通用函数，只是数学函数。它们执行快速的元素功能。当对NumPy数组执行简单的算术操作时，它们会自动调用，因为它们充当NumPy ufuncs的包装器。

例如，当使用“+”添加两个NumPy数组时，NumPy ufunc add()会在场景后面自动调用，并悄悄地发挥其魔力：

a = [1,2,3,4,5]
b = [6,7,8,9,10]
%timeit a+b

a = np.arange(1,6)
b = np.arange(6,11)
%timeit a+b

你可以看到，在NumPy ufuncs的帮助下，两个数组的相同添加是如何在更短的时间内完成的！

用NumPy数组计算

下面是一些最重要和最有用的操作，你将需要在你的NumPy数组上执行这些操作。

NumPy数组的基本运算

基本的算术运算可以很容易地在NumPy数组上执行。要记住的重要一点是，这些简单的算术运算符号只是作为NumPy ufuncs的包装器。

print('Subtract :',a-5)
print('Multiply :',a*5)
print('Divide :',a/5)
print('Power :',a**2)
print('Remainder :',a%5)

Subtract : [-4 -3 -2 -1 0]
Multiply : [ 5 10 15 20 25]
Divide : [0.2 0.4 0.6 0.8 1. ]
Power : [ 1 4 9 16 25]
Remainder : [1 2 3 4 0]

平均值、中位数和标准差

要查找NumPy数组的平均值和标准偏差，请使用mean()、std()和median()方法：

a = np.arange(5,15,2)
print('Mean :',np.mean(a))
print('Standard deviation :',np.std(a))
print('Median :',np.median(a))

Mean : 9.0
Standard deviation : 2.8284271247461903
Median : 9.0

最小最大值及其索引

使用Min()和Max()方法可以轻松找到ndarray中的Min和Max值：

a = np.array([[1,6],
[4,3]])
# 最小值
print('Min :',np.min(a,axis=0))
# 最大值
print('Max :',np.max(a,axis=1))

Min : [1 3]
Max : [6 4]

还可以使用argmin()和argmax()方法轻松确定ndarray中沿特定轴的最小值或最大值的索引：

a = np.array([[1,6,5],
[4,3,7]])
# 最小值
print('Min :',np.argmin(a,axis=0))
# 最大值
print('Max :',np.argmax(a,axis=1))

Min : [0 1 0]
Max : [1 2]

让我给你把输出分解一下。第一列的最小值是该列的第一个元素。对于第二列，它是第二个元素。对于第三列，它是第一个元素。

类似地，你可以确定最大值的输出指示什么。

在NumPy数组中排序

对于任何程序员来说，任何算法的时间复杂性都是最重要的。排序是一项重要且非常基本的操作，作为一名数据科学家，你可能每天都会用到它。因此，采用一种时间复杂度最小的排序算法是非常重要的。

当谈到排序数组元素时，NumPy库有一系列排序函数，可用于对数组元素进行排序。当你使用sort()方法时，它已经为你实现了快速排序、堆排序、合并排序和时间排序：

a = np.array([1,4,2,5,3,6,8,7,9])
np.sort(a, kind='quicksort')

array([1, 2, 3, 4, 5, 6, 7, 8, 9])

你甚至可以沿着你想要的任何轴对数组进行排序：

a = np.array([[5,6,7,4],
              [9,2,3,7]])# 沿列排序
print('Sort along column :','\n',np.sort(a, kind='mergresort',axis=1))
# 沿行排序
print('Sort along row :','\n',np.sort(a, kind='mergresort',axis=0))

Sort along column : 
 [[4 5 6 7]
 [2 3 7 9]]
Sort along row : 
 [[5 2 3 4]
 [9 6 7 7]]

NumPy数组和图像

NumPy数组在存储和操作图像数据方面有着广泛的用途。但图像数据到底是什么呢？

图像由以数组形式存储的像素组成。每个像素的值介于0到255之间–0表示黑色像素，255表示白色像素。

彩色图像由三个二维数组组成，每个彩色通道一个：红色、绿色和蓝色，背靠背放置，从而形成三维数组。数组中的每个值构成一个像素值。因此，数组的大小取决于每个维度上的像素数。

请看下图：

Python可以使用scipy.misc.imread()方法（SciPy库中的方法）。当我们输出它时，它只是一个包含像素值的三维数组：

import numpy as np
import matplotlib.pyplot as plt
from scipy import misc

# 读取图像
im = misc.imread('./original.jpg')
# 图像
im

array([[[115, 106,  67],
        [113, 104,  65],
        [112, 103,  64],
        ...,
        [160, 138,  37],
        [160, 138,  37],
        [160, 138,  37]],

       [[117, 108,  69],
        [115, 106,  67],
        [114, 105,  66],
        ...,
        [157, 135,  36],
        [157, 135,  34],
        [158, 136,  37]],

       [[120, 110,  74],
        [118, 108,  72],
        [117, 107,  71],
        ...,

我们可以检查这个NumPy数组的形状和类型：

print(im.shape)
print(type(type))

(561, 997, 3)
numpy.ndarray

现在，由于图像只是一个数组，我们可以使用本文中介绍的数组函数轻松地对其进行操作。比如，我们可以使用np.flip()方法：

# 翻转
plt.imshow(np.flip(im, axis=1))

或者你可以规范化或更改像素值的范围。这有时对更快的计算很有用。

im/255

array([[[0.45098039, 0.41568627, 0.2627451 ],
        [0.44313725, 0.40784314, 0.25490196],
        [0.43921569, 0.40392157, 0.25098039],
        ...,
        [0.62745098, 0.54117647, 0.14509804],
        [0.62745098, 0.54117647, 0.14509804],
        [0.62745098, 0.54117647, 0.14509804]],

       [[0.45882353, 0.42352941, 0.27058824],
        [0.45098039, 0.41568627, 0.2627451 ],
        [0.44705882, 0.41176471, 0.25882353],
        ...,
        [0.61568627, 0.52941176, 0.14117647],
        [0.61568627, 0.52941176, 0.13333333],
        [0.61960784, 0.53333333, 0.14509804]],

       [[0.47058824, 0.43137255, 0.29019608],
        [0.4627451 , 0.42352941, 0.28235294],
        [0.45882353, 0.41960784, 0.27843137],
        ...,
        [0.6       , 0.52156863, 0.14117647],
        [0.6       , 0.52156863, 0.13333333],
        [0.6       , 0.52156863, 0.14117647]],

       ...,

请记住，这是使用我们在文章中看到的ufuncs和广播的相同概念！

当你使用神经网络对图像进行分类时，你可以做更多的事情来操作你的图像。

结尾

我们在这篇文章中涉及了很多方面。希望你对NumPy数组的使用非常熟悉，并且非常热衷于将其融入到日常的分析任务中。

要了解更多关于任何NumPy函数的信息，请查看他们的官方文档，在那里你可以找到每个函数的详细描述。

文档链接：https://numpy.org/doc/

原文链接：https://www.analyticsvidhya.c...

欢迎关注磐创AI博客站：
http://panchuang.net/

sklearn机器学习中文官方文档：
http://sklearn123.com/

欢迎关注磐创博客资源汇总站：
http://docs.panchuang.net/