Numpy数组学习笔记

1、二维数组索引访问例子：

# 产生1-9的数
In [28]: arr2d = np.arange(1,10).reshape((3,3))

In [29]: arr2d
Out[29]: 
            array([[1, 2, 3],
                     [4, 5, 6],
                     [7, 8, 9]])
# 访问第一行数据
In [30]: arr2d[0]
Out[30]: array([1, 2, 3])
# 访问第一行第二个数据
In [31]: arr2d[0][1]
Out[31]: 2

In [32]: arr2d[0,1]
Out[32]: 2

2、X[:,0]是numpy中数组的一种写法，表示对一个二维数组，取该二维数组第一维中的所有数据，第二维中取第0个数据，直观来说，X[:,0]就是取所有行的第0个数据, X[:,1] 就是取所有行的第1个数据。

3、argmax返回最大数的索引。argmax有一个参数axis，默认是0，表示第几维的最大值。例子：

import numpy as np
a = np.array([[1, 5, 5, 2],
              [9, 6, 2, 8],
              [3, 7, 9, 1]])
print(np.argmax(a, axis=0))