向DataFrame中添加一列或多列的方法

最新推荐文章于 2024-07-16 12:34:52 发布

木头生辉

最新推荐文章于 2024-07-16 12:34:52 发布

阅读量1.2w

点赞数 6

分类专栏： python 文章标签： python loc函数 assign函数 concat函数 reindex函数

本文链接：https://blog.csdn.net/QQ315806208/article/details/113876882

版权

python 专栏收录该内容

39 篇文章 0 订阅

订阅专栏

这篇博客介绍了使用Pandas库在DataFrame中添加新列的四种方法：直接赋值、concat函数、reindex函数和assign函数。通过实例代码展示了如何简便地扩展数据框的列，包括直接为特定列添加数据、合并数据框以及利用现有数据进行重新索引和新增计算列。

摘要由CSDN通过智能技术生成

一、直接添加

代码：

import pandas as pd

test1 = pd.DataFrame([[1, 2, 3, 4], [2, 3, 4, 5], [3, 4, 5, 6], [4, 5, 6, 7]],columns=list('ABCD'))  #
print(test1)
test1['E']=[6,7,8,9]
print(test1)

二、通过concat函数添加

import pandas as pd

test1 = pd.DataFrame([[1, 2, 3, 4], [2, 3, 4, 5], [3, 4, 5, 6], [4, 5, 6, 7]],columns=list('ABCD'))  #
print(test1)
test2=pd.DataFrame([8,9,10,11],columns=list('F'))
test1=pd.concat([test1,test2],axis=1)
print(test1)

三、reindex函数添加

import pandas as pd

test1 = pd.DataFrame([[1, 2, 3, 4], [2, 3, 4, 5], [3, 4, 5, 6], [4, 5, 6, 7]],columns=list('ABCD'))  #
print(test1)

test3=test1.reindex(columns=list('MABCDN'),fill_value=1)
print(test3)

四、assign函数

import pandas as pd
import numpy as np

df = pd.DataFrame({'A':range(1, 6), 'B':np.random.randn(5)})
print(df)

df01=df.assign(dd=range(5,10))
print(df01)

五、loc函数添加

import pandas as pd

test1 = pd.DataFrame([[1, 2, 3, 4], [2, 3, 4, 5], [3, 4, 5, 6], [4, 5, 6, 7]],columns=list('ABCD'))  # 自己输入
print(test1)

test2=[False,False,True,False]
test2=pd.Series(test2)

test1.loc[test2,'E']=1
print(test1)