Pandas小技巧:在特定行后面增加一行数据
# 定义一个insert函数
def insert(df, i, df_add):
# 指定第i行插入一行数据
df1 = df.iloc[:i, :]
df2 = df.iloc[i:, :]
df_new = pd.concat([df1, df_add, df2], ignore_index=True)
return df_new
data = {
'name':['li', 'gg', 'zz'],
'age':[20, 21, 22],
'height':[170, 178, 174]}
df = pd.DataFrame(data)
print('df:')
print(df)
df_add = pd.DataFrame({'name':['yy'], 'age':[25], 'height':[168]})
# 在第2行插入一条新的数据
df_new = insert(df, 1, df_add)
print('df_new:')
print(df_new)