numpy总结

日常总结

numpy.array

import numpy as np
a=np.array([[0,1,2,3],
[4,5,6,7],
[0,1,2,3],
[4,5,6,7]])

mask=np.array([1,0,1,0],dtype=np.bool)

a[mask,2]

np.array()表示定义数组为bool类型，a[mask,2]中mask表示行数，为1的时候输出对应的结果。

numpy.random.random

import numpy as np
a = np.random.random((2,4))
print(a)
print(np.sum(a))
>>[[0.78866847 0.80411652 0.60990296 0.21026845]
 [0.11090346 0.25575255 0.75530674 0.8929869 ]]
4.427906039050992

表示随机返回一个两行四列的矩阵

numpy.nonzero

B = np.array([[3,5,9],
              [4,