ndarry的索引切片与迭代

  • 索引: ndarry数组的索引和python列表的索引相同, 都是用[index]表示.
  • 切片: 通过对每个以逗号分隔的维度执行单独的切片, 你可以对多维数组进行切片.

因此, 对于2D数组，我们的第一片定义了行的切片, 第二片定义了列的切片.
注意, 只需输入数字就可以指定行或列. 上面的第一个示例从数组中选择第0列.
下面的图表说明了给定的示例切片是如何进行工作的:

01

而迭代多维数组是相对于第一个轴完成的, 如果想要对数组中的每个元素执行操作, 可以使用 flat 属性, 该属性是数组中所有元素的迭代器:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
>>> import numpy as np
>>> a = np.arange(6).reshape(3, 2)
>>> a
array([[0, 1],
       [2, 3],
       [4, 5]])
>>> for i in a:
        print(i)

[0 1]
[2 3]
[4 5]
>>> for i in a.flat:
        print(i)

0
1
2
3
4
5

ndarry的形状和维度变换

ndarry的形状和维度变换有以下这些常用函数:

函数 说明
ndarry.reshape(shape) 不改变原ndarry元素并返回一个内容一样的shape形状的ndarry
ndarry.resize(shape) 改变原ndarry元素并使其与shape形状相同, 无返回值
ndarry.swapaxes(ax1, ax2) 返回ndarry的n个维度中的两个维度调换后的ndarry且不改变原ndarry
ndarry.flatten() 返回一个原ndarry降一维的ndarry
ndarry.astype(type) 返回一个新的数据类型为type的ndarry
ndarry.tolist() 返回ndarry转换而成的的列表

例如:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
>>> import numpy as np
>>> a = np.arange(6)
>>> a
array([0, 1, 2, 3, 4, 5])
>>> b = a.reshape((3, 2))
>>> b
array([[0, 1],
       [2, 3],
       [4, 5]])
>>> a.resize((3, 2))
>>> a
array([[0, 1],
       [2, 3],
       [4, 5]])
>>> np.swapaxes(a, 0, 1)
array([[0, 2, 4],
       [1, 3, 5]])
>>> a.flatten()
array([0, 1, 2, 3, 4, 5])
>>> a.astype(np.float)
array([[0., 1.],
       [2., 3.],
       [4., 5.]])
>>> a.tolist()
[[0, 1], [2, 3], [4, 5]]

还可以将不同数组叠加在一起, 或者将一个数组拆分:

函数 说明
np.stack(arrays, axis=0) 沿在aixs轴将n个形状完全相同的数组合并
np.vstack(tup) 在竖轴将n个不同数组堆叠, tup为一系列横轴数目相同的数组
np.hstack(tup) 在横轴将n个不同数组堆叠, tup为一系列竖轴数目相同的数组
np.dstack(tup) 在Z轴将n个不同数组堆叠, tup为一系列Z轴数目相同的数组
np.r_[] 沿横轴轴叠加数字创建数组
np.c_[] 沿竖轴叠加数字创建数组
np.split(a, indices_or_sections, axis=0) 在数组a的axis轴上拆分数组, 可以指定要返回的均匀划分的数组数量, 或指定要在其后进行划分的列
np.vsplit(a, indices_or_sections) vsplit与axis=0时的split相同
np.hsplit(a, indices_or_sections) hsplit与axis=1时的split相同
np.dsplit(a, indices_or_sections) dsplit与axis=2时的split相同

合并:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
>>> import numpy as np
>>> a = np.zeros((2, 3))
>>> a
array([[0., 0., 0.],
       [0., 0., 0.]])
>>> b = np.ones((2, 3))
>>> b
array([[1., 1., 1.],
       [1., 1., 1.]])
>>> np.stack((a, b))
array([[[0., 0., 0.],
        [0., 0., 0.]],

       [[1., 1., 1.],
        [1., 1., 1.]]])
>>> np.stack((a, b), axis=1)
array([[[0., 0., 0.],
        [1., 1., 1.]],

       [[0., 0., 0.],
        [1., 1., 1.]]])
>>> c = np.ones((1, 3))
>>> c
array([[1., 1., 1.]])
>>> np.vstack((a, c))
array([[0., 0., 0.],
       [0., 0., 0.],
       [1., 1., 1.]])
>>> d = np.ones((2, 1))
>>> d
array([[1.],
       [1.]])
>>> np.hstack((a, d))
array([[0., 0., 0., 1.],
       [0., 0., 0., 1.]])
>>> np.r_[1:4, 0, 4]
array([1, 2, 3, 0, 4])
>>> np.c_[1:4, 3:6, 5:8]
array([[1, 3, 5],
       [2, 4, 6],
       [3, 5, 7]])

需要注意, 使用stack的时候, 数组之间shape必须完全相同, vstack要求横轴(0轴)元素数目相同, hstack要竖轴(1轴)元素数目相同, dstack要求深度元素数目相同, 此时才能合并, 不然无法合并.

拆分:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
>>> import numpy as np
>>> a = np.arange(12).reshape(3, 4)
>>> a
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11]])
>>> np.split(a, 3)
[array([[0, 1, 2, 3]]), array([[4, 5, 6, 7]]), array([[ 8,  9, 10, 11]])]
>>> np.split(a, 2, axis=1)
[array([[0, 1],
       [4, 5],
       [8, 9]]),
 array([[ 2,  3],
       [ 6,  7],
       [10, 11]])]
np.split(a, (1, 2), axis=1)
[array([[0],
       [4],
       [8]]),
 array([[1],
       [5],
       [9]]),
 array([[ 2,  3],
       [ 6,  7],
       [10, 11]])]

通过例子可以看出, 当axis=0时, 划分的数组横轴(0轴)数目相同, 当axis=1时, 划分的数组竖轴(1轴)数目相同. 当指定其后拆分的行或列时, 会按照类似于切片的拆分, 如例子中的np.split(a, (1, 2), axis=1), 类似于将列像[:1], [1:2], [2:]这样切片.

ndarry与标量之间的计算

ndarry与标量之间的计算有以下这些常用函数:

函数 说明
ndarry.mean() 返回ndarry所有元素的平均值
np.fabs(x) 对 ndarry x的所有元素取绝对值, 不改变原ndarry的值
np.sqrt(x) 对 ndarry x的所有元素取平方根, 不改变原ndarry的值
np.square(x) 对 ndarry x的所有元素取平方, 不改变原ndarry的值
np.log(x) 对 ndarry x的所有元素取对数, 不改变原ndarry的值
np.ceil(x) 对 ndarry x的所有元素取ceiling值, 不改变原ndarry的值
np.floor(x) 对 ndarry x的所有元素取floor值, 不改变原ndarry的值
np.rint(x) 对 ndarry x的所有元素进行四舍五入, 不改变原ndarry的值
np.modf(x) 将 ndarry x的所有元素的整数部分和小数部分以两个独立数组的形式返回
np.sin(x) 对 ndarry x的所有元素取sin值, 不改变原ndarry的值
np.exp(x) 对 ndarry x的所有元素取指数值, 不改变原ndarry的值
np.sign(x) 对 ndarry x的所有元素取符号值1(+), 0(0), -1(-) 不改变原ndarry的值

例如:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
>>> import numpy as np
>>> a = np.linspace(-11, 11, 12).reshape((3, 4))
>>> a
array([[-11.,  -9.,  -7.,  -5.],
       [ -3.,  -1.,   1.,   3.],
       [  5.,   7.,   9.,  11.]])
>>> a.mean()
0.0
>>> np.sign(a)
array([[-1., -1., -1., -1.],
       [-1., -1.,  1.,  1.],
       [ 1.,  1.,  1.,  1.]])
>>> b = np.fabs(a)
>>> b
array([[11.,  9.,  7.,  5.],
       [ 3.,  1.,  1.,  3.],
       [ 5.,  7.,  9., 11.]])
>>> np.sqrt(b)
array([[3.31662479, 3.        , 2.64575131, 2.23606798],
       [1.73205081, 1.        , 1.        , 1.73205081],
       [2.23606798, 2.64575131, 3.        , 3.31662479]])
>>> np.square(b)
array([[121.,  81.,  49.,  25.],
       [  9.,   1.,   1.,   9.],
       [ 25.,  49.,  81., 121.]])
>>> c = np.log(b)
>>> c
array([[2.39789527, 2.19722458, 1.94591015, 1.60943791],
       [1.09861229, 0.        , 0.        , 1.09861229],
       [1.60943791, 1.94591015, 2.19722458, 2.39789527]])
>>> np.ceil(c)
array([[3., 3., 2., 2.],
       [2., 0., 0., 2.],
       [2., 2., 3., 3.]])
>>> np.floor(c)
array([[2., 2., 1., 1.],
       [1., 0., 0., 1.],
       [1., 1., 2., 2.]])
>>> np.rint(c)
array([[2., 2., 2., 2.],
       [1., 0., 0., 1.],
       [2., 2., 2., 2.]])
>>> np.modf(c)
(array([[0.39789527, 0.19722458, 0.94591015, 0.60943791],
       [0.09861229, 0.        , 0.        , 0.09861229],
       [0.60943791, 0.94591015, 0.19722458, 0.39789527]]),
 array([[2., 2., 1., 1.],
       [1., 0., 0., 1.],
       [1., 1., 2., 2.]]))
>>> np.sin(c)
array([[0.6770137 , 0.81012663, 0.93046593, 0.99925351],
       [0.89057704, 0.        , 0.        , 0.89057704],
       [0.99925351, 0.93046593, 0.81012663, 0.6770137 ]])
>>> np.exp(c)
array([[11.,  9.,  7.,  5.],
       [ 3.,  1.,  1.,  3.],
       [ 5.,  7.,  9., 11.]])

ndarry之间的比较

ndarry之间可以使用 < > <= >= == != 进行比较, 此时返回ndarry数组, 类型为bool
除此之外还有以下方法可以对ndarry之间进行比较:

函数 说明
np.maximum(x, y) np.fmax(x, y) ndarry对位元素最大值的计算
np.minimum(x, y) np.fmin(x, y) ndarry对位元素最小值的计算
np.mod(x, y) ndarry对位元素相除余数的计算, 即对位元素取模
np.copysign(x, y) 将ndarry y中各元素的符号赋给ndarry x中的对位元素

例如:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
>>> import numpy as np
>>> a = np.linspace(12, 23, 12).reshape(3, 4)
>>> a
array([[12., 13., 14., 15.],
       [16., 17., 18., 19.],
       [20., 21., 22., 23.]])
>>> b = np.linspace(1, 34, 12).reshape(3, 4)
>>> b
array([[ 1.,  4.,  7., 10.],
       [13., 16., 19., 22.],
       [25., 28., 31., 34.]])
>>> a > b
array([[ True,  True,  True,  True],
       [ True,  True, False, False],
       [False, False, False, False]])
>>> np.maximum(a, b)
array([[12., 13., 14., 15.],
       [16., 17., 19., 22.],
       [25., 28., 31., 34.]])
>>> np.mod(a, b)
array([[ 0.,  1.,  0.,  5.],
       [ 3.,  1., 18., 19.],
       [20., 21., 22., 23.]])