python numpy 基本函数

最新推荐文章于 2024-07-31 11:35:11 发布

CYummy

最新推荐文章于 2024-07-31 11:35:11 发布

阅读量349

点赞数

分类专栏： python numpy

本文链接：https://blog.csdn.net/CYummy/article/details/88787709

版权

本文介绍了numpy库的一些基本函数，包括numpy.prod用于计算数组元素乘积，numpy.diag处理对角线元素，numpy.concatenate实现数组拼接，numpy.cumsum计算累积和，以及numpy.argmax找到最大值的索引。通过实例展示了每个函数的使用方法，帮助理解numpy的功能。

摘要由CSDN通过智能技术生成

１、numpy.prod(a, axis, dtype, keepdims)

　　返回指定轴上的数组元素的乘积。

eg:

>>> c=np.arange(12).reshape(3,4)
>>> c
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11]])
>>> c.prod(axis=0)
array([  0,  45, 120, 231])
>>> c.prod(axis=1)
array([   0,  840, 7920])

2、numpy.diag(array)

array是一个1维数组时，结果形成一个以一维数组为对角线元素的矩阵

array是一个二维矩阵时，结果输出矩阵的对角线元素

eg:

>>> import numpy as np
>>> a=np.array([1,2,3])
>>> a
array([1, 2, 3])
>>> b=np.arange(9).reshape(3,3)
>>> b
array([[0, 1, 2],
       [3, 4, 5],
       [6, 7, 8]])
>>> b.shape
(3, 3)
>>> np.diag(b)
array([0, 4, 8])
>>> np.diag(a)
array([[1, 0, 0],
       [0, 2, 0],
       [0, 0, 3]])

3、numpy.concatenate((array1,array2