一些Numpy用法(一)

1.考虑一个(6,7,8)形状数组，第100个元素的索引(x,y,z)是多少

>>>print(np.unravel_index(99,(6,7,8)))

(1, 5, 3)

2.使用tile函数创建一个棋盘格8x8矩阵

>>>Z = np.tile( np.array([[0,1],[1,0]]), (4,4))
>>>print(Z)

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

3.创建一个自定义dtype，将颜色描述为四个无符号字节(RGBA)

>>>color = np.dtype([("r", np.ubyte, 1),
                  ("g", np.ubyte, 1),
                  ("b", np.ubyte, 1),
                  ("a", np.ubyte, 1)])
>>>print(color)

[('r', 'u1'), ('g', 'u1'), ('b', 'u1'), ('a', 'u1')]

4.5x3矩阵乘以3x2矩阵

>>>Z = np.dot(np.ones((5,3)), np.ones((3,2)))
>>>print(Z)

# Alternative solution, in Python 3.5 and above
>>>Z = np.ones((5,3)) @ np.ones((3,2))

[[3. 3.]
 [3. 3.]
 [3. 3.]
 [3. 3.]
 [3. 3.]]

5.看下面两个结果

>>>print(sum(range(5),-1))
>>>improt numpy as np
>>>print(np.sum(range(5),-1))

9  10

np.sum的第二个参数是维度,这里只有一维,即1+2+3+4=10