本文完整代码及附件已上传至我的
Github
仓库https://github.com/CNFeffery/DataScienceStudyNotes
大家好我是费老师,今天要给大家介绍的Python
框架叫做ibis
,没错,跟著名连锁酒店宜必思同名,其作者是创造了pandas
、Arrow
等著名框架的Wes McKinney。
ibis
的核心理念是用同一套数据框操作API,统一操纵各种主流的数据运算框架,使得用户可以更轻松更一致的构建自己的数据分析处理过程,而不是囿于不同框架之间千差万别的用法,下面我们就来一起学习其基础用法😉~
为了方便演示,我们这里使用conda
或mamba
创建新的虚拟环境,来安装试用ibis
,以mamba
为例,在终端中执行下列命令,完成演示用虚拟环境的创建及激活,并安装ibis
最基础的相关模块+示例数据集模块:
mamba create -n ibis-demo python=3.9 -y
mamba activate ibis-demo
pip install ibis-framework[duckdb,examples]
验证是否安装成功:
python -c "import ibis;print(ibis.__version__)"
可以看到,我们完成了对ibis
的安装,当前版本为9.0.0
:
ibis
本身不直接执行分析计算,当我们针对目标数据编写好对应计算逻辑的ibis
代码后,实际执行时其底层会将计算逻辑自动转换到当前的计算后端中。目前ibis
支持超过20种计算后端,均为当前单机分析、分布式分析领域的主流框架:
这使得基于ibis
构建的数据分析工作流,可以在不修改代码的情况下,轻松扩展应用于任意的计算后端。
譬如,ibis
默认使用DuckDB
作为后端进行单机分析运算,你可以在本机上使用部分数据构建并验证业务分析逻辑后,再套用相同的代码,切换计算后端到诸如ClickHouse
上执行分布式运算,这一特性也是ibis
便携性的体现。
举个简单的例子,首先我们基于pandas
生成具有一千万行记录的示例数据并导出为parquet
格式:
import numpy as np
import pandas as pd
# 生成测试用数据并导出为parquet格式
(
pd
.DataFrame(
{
'类别': np.random.choice(list('abcdef'), 10000000),
'数值': np.random.uniform(0, 100, 10000000)
}
)
.assign(数值=lambda df: df['数值'].round(3))
.to_parquet('demo.parquet')
)
接下来我们先使用默认的DuckDB
后端,在ibis
中执行一些示例运算,注意其运算耗时:
接着,我们切换计算后端为pandas
,执行完全一样的计算代码,可以看到计算耗时陡增,毕竟pandas
处理上千万行数据性能要远逊于DuckDB
,这个例子体现出ibis
强大的代码便携性:
ibis
中另一个非常强大的功能,是其可以充分结合Python
代码和SQL
代码来开展分析工作,譬如,你可以将ibis
分析代码直接转换为SQL
语句:
也可以直接执行SQL
语句开展分析:
这使得ibis
可以充分联结各类最先进的计算后端,帮助分析师轻松维护业务分析逻辑。
无论你原先在使用pandas
、SQL
还是R
进行数据分析,ibis
官网文档中都提供了非常友好的使用指南,确保你可以无痛的迁移使用ibis
:
ibis
正处于高速迭代发展阶段,其代码仓库几乎每天都在进行新的提交活动,欢迎进行⭐支持:https://github.com/ibis-project/ibis
更多相关内容,请移步其官网学习更多:https://ibis-project.org/
以上就是本文的全部内容,欢迎在评论区与我们进行讨论~