Numpy常用的功能

一维张量的基本属性

import numpy as np
a = np.array([1, 5, 3, 9, 2])
print(a.mean()) #求平均值
print(a.max())
print(a.min())
print(a.argmax())   #输出最大元素的下标，从0开始
print(a*2)
print(a.ndim)

二维张量的基本属性

import numpy as np
a = np.array([[1, 2, 3],
              [7, 3, 4],
              [5, 0, 1]]) #几维张量就有几个中括号
print(a.shape)
print(a.argmax()) #输出的结果是一个平铺后的下标，从0开始
print(a.flatten()) #平铺为一行
print(a.reshape(9, 1))

来源：CSDN

作者：skyline_ss

链接：https://blog.csdn.net/skylinesssss/article/details/103460677