Python怎么往数组中添加列


既然是数组,这里我们就用numpy的array来举例。

创建一个简单的数组

>>> import numpy as np
>>> x=np.array([[1,2],[3,4]]).T

>>> x
array([[1, 3],
       [2, 4]])
  • .T:相当于数学的转置的意思,这里我想让数据显示为垂向增加,所以用这个方法。

新的列

比如我们有一列新的数据:

>>> new_col = [[5],[6]]
  • 注意这里[[5],[6]]的写法,两个数字是在同一列上的,所以分别用方括号括起来并用逗号隔开

numpy.append()函数

python的列表一样,numpy的数组也有append()函数,使用该函数向数组中添加新的数据:

>>> new_x = np.append(x,new_col,axis=1)
>>> new_x
array([[1, 3, 5, 5],
       [2, 4, 6, 6]])

np.append(x,new_col,axis=1):

  • x:原有的数组
  • new_col:要添加的数据
  • axis=1:因为要添加的是列,所以坐标轴为1(y轴),添加行的话则是0(x轴),和pandas的dataframe是一样的,可以看看这里的图解:pandas 数据框/帧

相关

Python 数组array