Python中对numpy库的学习（五）常用函数2 组合与分割

目录

一、概念

二、组合

1、stack函数

2、concatenate函数

 3、flatten和reval方法

三、分割

1、split和array_split方法

---

一、概念

  组合：将数组连接在一起，形成一个新数组。

  分割：将数组中的元素分割成几个数组。

二、组合

1、stack函数

   作用：将形状相同的数组作为元素连接成一个新数组，结果会多加一维（外层加一层[  ]）。使用方法np.stack((a,b))，ab为数组名（可多个，注意括号）。

import numpy as np

a = np.array([1, 2])
b = np.array([3, 4])
c = np.stack((a, b))
print(c)
# [[1 2]
#  [3 4]]

a = np.array([[1, 2], [3, 4]])
b = np.array([[5, 6], [7, 8]])
c = np.stack((a, b))
print(c)
# [[[1 2]
#  [3 4]]
# 
#  [[5 6]
#   [7 8]]]
print('c.shape = ', c.shape)    # c.shape =  (2, 2, 2)
print('c.ndim = ', c.ndim)    # c.ndim =  3

2、concatenate函数

  conc