python: numpy -- zeros,ones,eye函数

  • zeros():可以用来构造全0数组

如果是一维的,则需要一个括号,里面的数字代表数组的长度,如zeros(3)表示创建1行3列全为0的数组

如果是多维的,则要有两个括号,里面的维度用tuple来表示,如zeros((3,3))表示创建3行3列全为0的数组

>>> zeros(3)
array([ 0.,  0.,  0.])
>>> zeros((3,3))
array([[ 0.,  0.,  0.],
       [ 0.,  0.,  0.],
       [ 0.,  0.,  0.]])
  • ones(): 可以用来构造全1的数组

如果是多维的,则要有两个括号,里面的维度用tuple来表示,如a=ones((3,1))表示创建3行1列全为1的数组

如果是一维的,则需要一个括号,里面的数字代表数组的长度,如a=ones(3)表示创建1行3列全为1的数组

a=ones((3,1))
print("a:\n",a)
print("type(a):",type(a))
b=ones(3)
print("b:\n",b)
print("type(b):",type(b))


#运行结果如下:
a:
 [[1.]
 [1.]
 [1.]]
type(a): <class 'numpy.ndarray'>
b:
 [1. 1. 1.]
type(b): <class 'numpy.ndarray'>
>>> ones((3,3))
array([[ 1.,  1.,  1.],
       [ 1.,  1.,  1.],
       [ 1.,  1.,  1.]])
  • eye(): 创建一个正方的NxN单位矩阵(对角线为1,其余为0)
>>> eye(3)
array([[ 1.,  0.,  0.],
       [ 0.,  1.,  0.],
       [ 0.,  0.,  1.]])

 

猜你喜欢

转载自blog.csdn.net/jiangsujiangjiang/article/details/83659765
今日推荐