numpy函数

函数概览：

np.pad(array, pad_width, mode='constant', **kwargs) 
np.diag(v, k)
#仅列出我用到的参数

---

函数详解：

numpy.diag(v, k) #数组对角线的相关操作

---

参数解释:

  k:

• type:int
• 默认为0，表示主对角线，如果为1就是主对角线上方的第一条对角线，-1为主对角线下方的第一条对角线，以此类推…

  v:

• type:数组
• 如果是一维数组，直接作为第k条对角线，数组的维度由此得出。如果是二维数组，将其第k条对角线返回（k的含义如上k:所示）

---

返回值

  return:

• type:数组

---

示例：

#一维数组作为参数
Z = np.diag(np.arange(1,5),k=0)
Z

输出:
array([[1, 0, 0, 0],
[0, 2, 0, 0],
[0, 0, 3, 0],
[0, 0, 0, 4]])

#二维数组作为参数
Z = np.diag(np.arange(9),reshape(3,3),k=1)
Z

Z = [[0 1 2]
[3 4 5]
[6 7 8]] #主对角线上方第一条对角线为[1,5]
输出：
array([1, 5])

---

np.pad(array, pad_width, mode='constant', **kwargs) #填充数组的边缘，只讨论二维及以下
参数解释:

  array:

• type:数组
• 给定需要填充的数组

  pad_width:

• type:序列 or 数字
• 若为数字,表示填充的厚度，为1表示填充1个

  mode:

• type:字符串 or 函数
  填充模式

  constant_values:

• type:填充的值，可以是序列(如果pad_width是序列的话)

---

返回值

• type:数组

---

示例

#一维数组,前面用4填充两层，后面用6填充三层
a = [1, 2, 3, 4, 5]
np.pad(a, pad_width = (2, 3), mode = 'constant', constant_values=(4, 6))

array([4, 4, 1, 2, 3, 4, 5, 6, 6, 6])

#创建一个6*6的数组，外围由1包裹，里面全0
a = np.zeros([5,5])
np.pad(a,pad_width=1,mode='constant',constant_values=1)

array([[1., 1., 1., 1., 1., 1., 1.],
[1., 0., 0., 0., 0., 0., 1.],
[1., 0., 0., 0., 0., 0., 1.],
[1., 0., 0., 0., 0., 0., 1.],
[1., 0., 0., 0., 0., 0., 1.],
[1., 0., 0., 0., 0., 0., 1.],
[1., 1., 1., 1., 1., 1., 1.]])