使用字典在 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

相关文章