numpy的使用

1. 判断两个数组是否完全一致

   a = np.array([1,2,3])
   b = np.array([1,2,3])
   print((a==b).all())
   

2. list与ndarray转换

   # list -> ndarray
   np.array(list)
   # ndarray -> list
   arr.tolist()
   

3. bytes与ndarray转换

   arr = np.array([1,2,3])
   # ndarray -> bytes
   b = arr.tobytes()
   # bytes -> ndarray
   # 得到的是一维数组，如果之前是多维，需要手动reshape
   arr = np.frombuffer(b)
   

4. ndarray大于某一数值的统计

   # arr中大于1的个数
   count = np.sum(np.greater(arr, 1))
   

5. 添加维度

   arr = [1, 2, 3, 4, 5]
   arr = np.array(arr)
   # 输出为(5,)
   print(arr.shape)
   arr_1 = arr[np.newaxis, :]
   # 输出为(1, 5)
   print(arr_1.shape)
   arr_2 = arr[:, np.newaxis]
   # 输出为(5, 1)
   print(arr_2.shape)