Pandas:让你像写SQL一样做数据分析

1. 引言

Pandas是一个开源的Python数据分析库。Pandas把结构化数据分为了三类:

  • Series,1维序列,可视作为没有column名的、只有一个column的DataFrame;
  • DataFrame,同Spark SQL中的DataFrame一样,其概念来自于R语言,为多column并schema化的2维结构化数据,可视作为Series的容器(container);
  • Panel,为3维的结构化数据,可视作为DataFrame的容器;

DataFrame较为常见,因此本文主要讨论内容将为DataFrame。DataFrame的生成可通过读取纯文本、Json等数据来生成,亦可以通过Python对象来生成:

import pandas as pd
import numpy as np

df = pd.DataFrame({‘total_bill‘: [16.99, 10.34, 23.68, 23.68, 24.59],
                   ‘tip‘: [1.01, 1.66, 3.50, 3.31, 3.61],
                   ‘sex‘: [‘Female‘, ‘Male‘, ‘Male‘, ‘Male‘, ‘Female‘]})

对于DataFrame,我们可以看到其固有属性:

# data type of columns
print df.dtypes
# indexes
print df.index
# return pandas.Index
print df.columns
# each row, return array[array]
print df.values
  • .index,为行索引
  • .columns,为列名称(label)
  • .dtype,为列数据类型

2. SQL操作

官方Doc给出了部分SQL的Pandas实现。在此基础上,本文给出了一些扩充说明。以下内容基于Python 2.7 + Pandas 0.18.1的版本。

select

SQL中的select是根据列的名称来选取;Pandas则更为灵活,不但可根据列名称选取,还可以根据列所在的position选取。相关函数如下:

  • loc,基于列label,可选取特定行(根据行index);
  • iloc,基于行/列的position;
print df.loc[1:3, [‘total_bill‘, ‘tip‘]]
print df.loc[1:3, ‘tip‘: ‘total_bill‘]
print df.iloc[1:3, [1, 2]]
print df.iloc[1:3, 1: 3]
  • at,根据指定行index及列label,快速定位DataFrame的元素;
  • iat,与at类似,不同的是根据position来定位的;
print df.at[3, ‘tip‘]
print df.iat[3, 1]
  • ix,为loc与iloc的混合体,既支持label也支持position;
print df.ix[1:3, [1, 2]]
print df.ix[1:3, [‘total_bill‘, ‘tip‘]]

此外,有更为简洁的行/列选取方式:

print df[1: 3]
print df[[‘total_bill‘, ‘tip‘]]
# print df[1:2, [‘total_bill‘, ‘tip‘]]  # TypeError: unhashable type

where

Pandas实现where filter,较为常用的办法为df[df[colunm] boolean expr],比如:

print df[df[‘sex‘] == ‘Female‘]
print df[df[‘total_bill‘] > 20]

# or
print df.query(‘total_bill > 20‘)

在where子句中常常会搭配and, or, in, not关键词,Pandas中也有对应的实现:

# and
print df[(df[‘sex‘] == ‘Female‘) & (df[‘total_bill‘] > 20)]
# or
print df[(df[‘sex‘] == ‘Female‘) | (df[‘total_bill‘] > 20)]
# in
print df[df[‘total_bill‘].isin([21.01, 23.68, 24.59])]
# not
print df[-(df[‘sex‘] == ‘Male‘)]
print df[-df[‘total_bill‘].isin([21.01, 23.68, 24.59])]
# string function
print df = df[(-df[‘app‘].isin(sys_app)) & (-df.app.str.contains(‘^微信\d+$‘))]

对where条件筛选后只有一行的dataframe取其中某一列的值,其两种实现方式如下:

total = df.loc[df[‘tip‘] == 1.66, ‘total_bill‘].values[0]
total = df.get_value(df.loc[df[‘tip‘] == 1.66].index.values[0], ‘total_bill‘)

distinct

drop_duplicates根据某列对dataframe进行去重:

df.drop_duplicates(subset=[‘sex‘], keep=‘first‘, inplace=True)

包含参数:

  • subset,为选定的列做distinct,默认为所有列;
  • keep,值选项{‘first‘, ‘last‘, False},保留重复元素中的第一个、最后一个,或全部删除;
  • inplace ,默认为False,返回一个新的dataframe;若为True,则返回去重后的原dataframe

group

group一般会配合合计函数(Aggregate functions)使用,比如:count、avg等。Pandas对合计函数的支持有限,有count和size函数实现SQL的count:

print df.groupby(‘sex‘).size()
print df.groupby(‘sex‘).count()
print df.groupby(‘sex‘)[‘tip‘].count()

对于多合计函数,

select sex, max(tip), sum(total_bill) as total
from tips_tb
group by sex;

实现在agg()中指定dict:

print df.groupby(‘sex‘).agg({‘tip‘: np.max, ‘total_bill‘: np.sum})

# count(distinct **)
print df.groupby(‘tip‘).agg({‘sex‘: pd.Series.nunique})

as

SQL中使用as修改列的别名,Pandas也支持这种修改:

# first implementation
df.columns = [‘total‘, ‘pit‘, ‘xes‘]
# second implementation
df.rename(columns={‘total_bill‘: ‘total‘, ‘tip‘: ‘pit‘, ‘sex‘: ‘xes‘}, inplace=True)

其中,第一种方法的修改是有问题的,因为其是按照列position逐一替换的。因此,我推荐第二种方法。

join

Pandas中join的实现也有两种:

# 1.
df.join(df2, how=‘left‘...)

# 2.
pd.merge(df1, df2, how=‘left‘, left_on=‘app‘, right_on=‘app‘)

第一种方法是按DataFrame的index进行join的,而第二种方法才是按on指定的列做join。Pandas满足left、right、inner、full outer四种join方式。

order

Pandas中支持多列order,并可以调整不同列的升序/降序,有更高的排序自由度:

print df.sort_values([‘total_bill‘, ‘tip‘], ascending=[False, True])

top

对于全局的top:

print df.nlargest(3, columns=[‘total_bill‘])

对于分组top,MySQL的实现(采用自join的方式):

select a.sex, a.tip
from tips_tb a
where (
    select count(*)
    from tips_tb b
    where b.sex = a.sex and b.tip > a.tip
) < 2
order by a.sex, a.tip desc;

Pandas的等价实现,思路与上类似:

# 1.
df.assign(rn=df.sort_values([‘total_bill‘], ascending=False)
          .groupby(‘sex‘)
          .cumcount()+1)    .query(‘rn < 3‘)    .sort_values([‘sex‘, ‘rn‘])

# 2.
df.assign(rn=df.groupby(‘sex‘)[‘total_bill‘]
          .rank(method=‘first‘, ascending=False))     .query(‘rn < 3‘)     .sort_values([‘sex‘, ‘rn‘])

replace

replace函数提供对dataframe全局修改,亦可通过where条件进行过滤修改(搭配loc):

# overall replace
df.replace(to_replace=‘Female‘, value=‘Sansa‘, inplace=True)

# dict replace
df.replace({‘sex‘: {‘Female‘: ‘Sansa‘, ‘Male‘: ‘Leone‘}}, inplace=True)

# replace on where condition
df.loc[df.sex == ‘Male‘, ‘sex‘] = ‘Leone‘

自定义

除了上述SQL操作外,Pandas提供对每列/每一元素做自定义操作,为此而设计以下三个函数:

  • map(func),为Series的函数,DataFrame不能直接调用,需取列后再调用;
  • apply(func),对DataFrame中的某一行/列进行func操作;
  • applymap(func),为element-wise函数,对每一个元素做func操作
print df[‘tip‘].map(lambda x: x - 1)
print df[[‘total_bill‘, ‘tip‘]].apply(sum)
print df.applymap(lambda x: x.upper() if type(x) is str else x)

3. 实战

环比增长

现有两个月APP的UV数据,要得到月UV环比增长;该操作等价于两个Dataframe left join后按指定列做减操作:

def chain(current, last):
    df1 = pd.read_csv(current, names=[‘app‘, ‘tag‘, ‘uv‘], sep=‘\t‘)
    df2 = pd.read_csv(last, names=[‘app‘, ‘tag‘, ‘uv‘], sep=‘\t‘)
    df3 = pd.merge(df1, df2, how=‘left‘, on=‘app‘)
    df3[‘uv_y‘] = df3[‘uv_y‘].map(lambda x: 0.0 if pd.isnull(x) else x)
    df3[‘growth‘] = df3[‘uv_x‘] - df3[‘uv_y‘]
    return df3[[‘app‘, ‘growth‘, ‘uv_x‘, ‘uv_y‘]].sort_values(by=‘growth‘, ascending=False)

差集

对于给定的列,一个Dataframe过滤另一个Dataframe该列的值;相当于集合的差集操作:

def difference(left, right, on):
    """
    difference of two dataframes
    :param left: left dataframe
    :param right: right dataframe
    :param on: join key
    :return: difference dataframe
    """
    df = pd.merge(left, right, how=‘left‘, on=on)
    left_columns = left.columns
    col_y = df.columns[left_columns.size]
    df = df[df[col_y].isnull()]
    df = df.ix[:, 0:left_columns.size]
    df.columns = left_columns
    return df
时间: 2024-08-26 00:56:45

Pandas:让你像写SQL一样做数据分析的相关文章

像写SQL一样编写Java数据应用-TinySqlDsl

前言话说企业应用,一般离不开数据库.要做数据库,可以有N种方案,比如:直接采用JDBC层自己封装下使用的,采用一些框架的,如:iBatis,Hiberate,Spring JDBC Template等等(这个太多了,因此不一一列举)的,这些方案也都在各自的领域展示了自己的特点,解决了相当部分的技术问题,并取得了相当好的应用效果. 但是不管是哪种方案,其优点和缺点往往也是连在一起的,究其原因是因为SQL和Java编程之间是割裂的,如果封装得不到位,做Java的人太难使用:如果封装得太多,在做一些用

Java开发者写SQL时常犯的10个错误

首页 所有文章 资讯 Web 架构 基础技术 书籍 教程 我要投稿 更多频道 » - 导航条 - 首页 所有文章 资讯 Web 架构 基础技术 书籍 教程 我要投稿 更多频道 » - iOS - Python - Android - Web前端 Java开发者写SQL时常犯的10个错误 2015/03/10 | 分类: 基础技术 | 0 条评论 | 标签: SQL 分享到:0 本文由 ImportNew - zer0Black 翻译自 jooq.欢迎加入翻译小组.转载请见文末要求. 我十分惊讶的

程序员老鸟写sql语句的经验之谈

做管理系统的,无论是bs结构的还是cs结构的,都不可避免的涉及到数据库表结构的设计,sql语句的编写等.因此在开发系统的时候,表结构设计是否合理,sql语句是否标准,写出的sql性能是否优化往往会成为公司衡量程序员技术水平的标准. 我们程序员不是dba,不需要时刻关注sql运行时间,想方设法优化表结构,存储空间,优化表读取速度等等,但是在开发系统时,时刻保持优良的写sql语句的作风是很有必要的,这关乎到个人在公司的声誉,嘿嘿,你懂的... 新来的程序员老鸟,在一个开发团队中,需要表现一下自己的水

Java AVL、BST编程作业代写、代做JDK实验

Java AVL.BST编程作业代写.代做JDK实验1 IntroductionNeeding a way to manage all your PlayStation friends, you decide to build a backendsystem for adding, removing and maintaining them. The idea is to organiseyour friends so you can search for individuals, search

java eclipse 作业代写、代做eclipse实验

java eclipse 作业代写.代做eclipse实验<java程序设计>课程实验报告 实验四 继承与多态一.实验目的及要求1.了解如何使用类及其成员的修饰符,掌握方法的继承.重载和覆盖.2.掌握类的继承关系和派生方法.3.掌握多态的概念与使用. 4.掌握接口的定义与使用.二.实验内容(一)类的继承性练习1.进一步理解继承的含义 新类可从现有的类中产生,并保留现有类的成员变量和方法并可根据需要对它们加以修改.新类还可添加新的变量和方法.这种现象就称为类的继承. 当建立一个新类时,不必写出全

大数据不就是写SQL吗?

[原创,转载请注明出处] 应届生小祖参加了个需求分析会回来后跟我说被产品怼了一句: "不就是写SQL吗,要那么久吗" 我去,欺负我小弟,这我肯定不能忍呀,于是我写了一篇文章发在了公司的wiki 贴出来给大家看看,省略了一些敏感的内容.当然内部版言辞也会温和一点,嘻嘻 在哪里写SQL? 这个问题高级点的问法是用哪种SQL引擎? SparkSQL.Hive.Phoenix.Drill.Impala.Presto.Druid.Kylin (这里的SQL引擎是广义的,大家不必钻牛角尖) 我用一

hive--构建于hadoop之上、让你像写SQL一样编写MapReduce程序

hive介绍 什么是hive? hive:由Facebook开源用于解决海量结构化日志的数据统计 hive是基于hadoop的一个数据仓库工具,可以将结构化的数据映射为数据库的一张表,并提供类SQL查询功能.本质就是将HQL(hive sql)转化为MapReduce程序 我们使用MapReduce开发会很麻烦,但是程序员很熟悉sql,于是hive就出现了,可以让我们像写sql一样来编写MapReduce程序,会自动将我们写的sql进行转化.但底层使用的肯定还是MapReduce. hive处理

springboot项目里,让tk-mybatis支持可以手写sql的mapper.xml文件

SpringBoot项目通常配合TKMybatis或MyBatis-Plus来做数据的持久化. 对于单表的增删改查,TKMybatis优雅简洁,无需像传统mybatis那样在mapper.xml文件里定义sql. 我们目前的项目呢,有一些数据分析的需求,涉及到多表关联.嵌套子查询等复杂的sql. 那么,TKMybatis是不是可以支持手写sql呢? 答案是yes! 我们知道,springboot集成tk-mybatis需添加2个依赖: <dependency> <groupId>t

夜黑风高的夜晚用SQL语句做了一些想做的事&#183;&#183;&#183;&#183;&#183;&#183;&#183;

     IT这条漫漫长路注定是孤独的,陪伴我们的只有那些不知冷暖的代码语句和被手指敲打的磨掉了键上的标识的键盘. 之所以可以继续坚持下去,是因为心中有一份永不熄灭的激情. 成功的路上让我们为自己带盐~~~~     想知道SQL语句的强大功能就先让我给你讲讲什么是SQL吧 SQL简介:        SQL能做什么?            专业人士说:在进行数据管理时,如果每次创建数据库,表或者从数据库中读取数据,都需要手动在SQL Server Management  Studio  中进行