使用字典在 Pandas 数据框中添加新列
pandasserver side programmingprogramming
Pandas 数据框是一种二维数据结构,即数据以表格形式排列在行和列中。可以使用 python 字典、列表和系列等创建。在本文中,我们将了解如何向现有数据框添加新列。
首先让我们使用 pandas 系列创建一个数据框。在下面的例子中,我们将一个 pandas 系列转换为一个只有一列的数据框,并给它一个列名 Month_no。
示例
import pandas as pd s = pd.Series([6,8,3,1,12]) df = pd.DataFrame(s,columns=['Month_No']) print (df)
输出
运行上述代码得到以下结果 −
Month_No 0 6 1 8 2 3 3 1 4 12
接下来,我们创建一个新的 Python 字典,其中包含月份名称,并使用来自 Pandas 系列的值作为字典的索引。然后,我们使用 map 函数将月份的字典与现有的数据框相加,以获得一个新列。map 函数负责将月份名称与字典的索引进行排列。
示例
import pandas as pd s = pd.Series([6,8,3,1,12]) df = pd.DataFrame(s,columns=['Month_No']) months = {6:'Jun', 8:'Aug', 3:'Mar', 1:'Jan',12:'Dec'} df['Month_Name'] = df['Month_No'].map(months) print (df)
输出
运行上述代码得到以下结果 −
Month_No Month_Name 0 6 Jun 1 8 Aug 2 3 Mar 3 1 Jan 4 12 Dec