Python中ndarray数组加入一行元素


ndarray

是numpy扩展库下的内容，用来表示多维数组(n-dimensions array)：

import numpy as np
a=np.array([[1,2],[3,4],[5,6]])

事先不知道数组的尺寸可以先定义一个空数组，再向其中加入元素：

b = np.array([])
# 筛选a的每一行中所有元素都大于2的行，并保存到b中
b = np.append(b, [row for row in a if min(row) > 2])

结果为：

b = [3. 4. 5. 6.]

再进行reshape操作即可实现目的：

# [-1,2]的2表示2列，-1表示行数未知，由数组而定 
b = np.reshape(b, [-1, 2])  

结果为：

b = [[3. 4. ],[5. 6.]]