numpy的基本用法(三)——numpy的索引

| |
numpy的基本用法(三)——numpy的索引

文章作者：Tyan
博客：noahsnail.com  |  CSDN  |  简书

本文主要是关于numpy的一些基本运算的用法。

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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
#!/usr/bin/env python
# _*_ coding: utf-8 _*_

import numpy as np

# Test 1
# 一维矩阵
a = np.arange(3, 15)
print a
# 输出矩阵的第三个元素
print a[2]

# Test 1 result
[ 3  4  5  6  7  8  9 10 11 12 13 14]
5

# Test 2
# 二维矩阵
a = np.arange(3, 15).reshape(3, 4)
print a
# 输出矩阵的第二行
print a[1]
# 输出矩阵的第一个元素
print a[0][0]
# 输出矩阵某个位置上的元素
print a[2][1]
print a[2, 1]
# 输出矩阵第三行的所有数字
# :代表整行或整列
print a[2, :]
# 输出矩阵第二行的前三个数，左开右闭
print a[1, 0:3]

# Test 2 result
[[ 3  4  5  6]
 [ 7  8  9 10]
 [11 12 13 14]]
[ 7  8  9 10]
3
12
12
[11 12 13 14]
[7 8 9]

# Test 3
# 迭代矩阵的行
for row in a:
    print row

# 迭代矩阵的列
for column in a.T:
    print column

# Test 3 result
[3 4 5 6]
[ 7  8  9 10]
[11 12 13 14]

[ 3  7 11]
[ 4  8 12]
[ 5  9 13]
[ 6 10 14]

# Test 4
# 矩阵展开
print a.flatten()
# 迭代矩阵的元素
for item in a.flat:
    print item

# Test 4 result
[ 3  4  5  6  7  8  9 10 11 12 13 14]
3
4
5
6
7
8
9
10
11
12
13
14

参考资料

  1. https://www.youtube.com/user/MorvanZhou
如果有收获，可以请我喝杯咖啡！