在指定位置添加一列值
加载数据
首先,我们需要先将需要的数据加载进来,这里使用了一个比较复杂的数据,总共是15列
import pandas as pd
import numpy as np
data = pd.read_csv('train.csv')
显示数据,及行,列值,使用shape
添加新列
columns = data.columns.tolist()
columns.insert(0,'add')
data = data.reindex(columns = columns)
data['add'] = np.ones(data.shape[0],dtype=np.int)
print(type(data))
print(data.shape)
这样添加的好处是,添加前的数据类型和添加后的一样,都是Dataframe类型
直接使用insert添加
import pandas as pd
import numpy as np
data = pd.read_csv('train.csv')
# print(data,data.shape)
data.insert(0,'add',1)
print(type(data),data.shape)
这里可以看到数据新增一列,并且数据类型不变。