Numpy

argmax用法

还是从一维数组出发.看下面的例子:

import numpy as np
a = np.array([2, 1, 2, 4, 6, 10])
print(np.argmax(a))
5

argmax返回的是最大数的索引.argmax有一个参数axis, 默认是0,表示第几维的最大值。

数组的转置乘法

import numpy as np
np.dot(np.ones(2),np.array([2,3]))
5.0
a = np.random.randn(2,2)

Out[31]: array([[-0.67949478,  1.67333199],
       [-0.90951512,  1.66326058]])

np.transpose(a)
Out[32]: array([[-0.67949478, -0.90951512],
       [ 1.67333199,  1.66326058]])