在Pandas的DataFrame中添加一行或者一列他匪,添加行有df.loc[]
以及df.append()
這兩種方法岖圈,添加列有df[]
和df.insert()
兩種方法弯洗, 下面對這幾種方法的使用進行簡單介紹亲轨。
一杠步、添加行
- 添加一行肠鲫,采用
loc[]
方法
# 構(gòu)造一個空的dataframe
import pandas as pd
df = pd.DataFrame(columns=['name','number'])
# 采用.loc的方法進行
df.loc[0]=['cat', 3] # 其中l(wèi)oc[]中需要加入的是插入地方dataframe的索引员帮,默認(rèn)是整數(shù)型
# 也可采用諸如df.loc['a'] = ['123',30]的形式
采用loc[]
方法多適用于對空的dataframe循環(huán)遍歷添加行,這樣索引可以從0開始直到數(shù)據(jù)結(jié)果导饲,不會存在索引沖突的問題捞高。
- 添加一行或合并兩個dataframe,采用
append()
方法
# 1. 采用append方法合并兩個dataframe
# 構(gòu)造兩個dataframe
df = pd.DataFrame([[1, 2], [3, 4]], columns=list('AB'))
df2 = pd.DataFrame([[5, 6], [7, 8]], columns=list('AB'))
# 合并 ignore_index設(shè)置為 True可以重新排列索引
df.append(df2, ignore_index=True)
A B
0 1 2
1 3 4
2 5 6
3 7 8
# 2. 采用append方法添加多行
df = pd.DataFrame(columns=['A'])
for i in range(5):
df = df.append({'A': i}, ignore_index=True)
df
A
0 0
1 1
2 2
3 3
4 4
# 同樣如果是遍歷添加多行帜消,有一種更高效的方法
pd.concat([pd.DataFrame([i], columns=['A']) for i in range(5)],
ignore_index=True)
A
0 0
1 1
2 2
3 3
4 4
二棠枉、添加列
- 新增一列,采用
df[]
方法直接在列上操作
# 新建一個dataframe
df = pd.DataFrame(columns=['name','number'], data=[['cat',3]])
df
name number
0 cat 3
# 添加一列泡挺,計算有多少條腿
df['leg'] = df['number'] * 4
# 添加一列辈讶,直接賦值有幾個頭
df['head'] = 1
df
name number leg head
0 cat 3 12 1
- 添加一列,采用
insert()
方法
# 使用方法是DataFrame.insert(loc, column, value, allow_duplicates=False)
# 即df.insert(添加列位置索引序號娄猫,添加列名贱除,數(shù)值,是否允許列名重復(fù))
df.insert(1, 'tail', 1, allow_duplicates=False)
df
name tail number leg head
0 cat 1 3 12 1
不過在使用insert的過程中發(fā)現(xiàn)454: DeprecationWarning: `input_splitter` is deprecated since IPython 7.0, prefer `input_transformer_manager`. status, indent_spaces = self.shell.input_splitter.check_complete(code)
這個提示媳溺,猜測是有別的地方出問題了月幌,還需要調(diào)試。
主要參考資料: