pandas实现excel中的数据透视表和Vlookup函数功能代码

  • Post category:Python

请看下面的示例说明。

pandas实现Excel中的数据透视表

在Excel中,我们可以通过数据透视表来方便地对数据进行分类汇总和分析。而在pandas中,也提供了类似的功能。下面是一个数据透视表的实现示例:

import pandas as pd

# 创建数据
data = {
    '省份': ['北京', '北京', '上海', '上海', '广东', '广东'],
    '城市': ['北京', '北京', '上海', '上海', '深圳', '广州'],
    '销售额': [1000, 2000, 3000, 4000, 5000, 6000]
}
df = pd.DataFrame(data)

# 对数据进行分组和汇总
table = pd.pivot_table(df, values='销售额', index=['省份'], columns=['城市'], aggfunc=sum)
print(table)

在这个示例中,我们创建了一个包含省份、城市、销售额三个字段的数据,接着使用pandas的pivot_table函数将数据进行分组和汇总。最后得到的结果就是一个数据透视表,其中省份作为行,城市作为列,销售额作为值,每个单元格中存储的是对应省份和城市的销售额之和。

pandas实现Excel中的Vlookup函数

除了数据透视表,Excel中还有一个常用的函数是Vlookup,可以根据指定的关键字查找对应的数据。同样地,在pandas中也提供了类似的功能。下面是一个Vlookup函数实现的示例:

import pandas as pd

# 创建数据
data = {
    '城市': ['北京', '上海', '广州', '深圳'],
    '人口': [2000, 3000, 4000, 5000],
    'GDP': [100, 200, 300, 400]
}
df1 = pd.DataFrame(data)

data2 = {
    '城市': ['北京', '上海', '广州', '深圳', '武汉', '重庆'],
    '经度': [116, 121, 113, 114, 114, 106],
    '纬度': [39, 31, 23, 22, 30, 29]
}
df2 = pd.DataFrame(data2)

# 使用merge函数合并两个表格
result = pd.merge(df1, df2, on='城市')
print(result)

在这个示例中,我们创建了两个数据表,一个表格中存储城市、人口和GDP三个数据,另一个表格中存储城市、经度和纬度三个数据。接着使用pandas的merge函数将两个表格根据城市这个关键字进行合并。这样我们就可以方便地查找某个城市的经度和纬度了。

以上两个示例涵盖了pandas实现Excel中数据透视表和Vlookup函数的基本用法,希望能对你有所帮助。