高效使用ndarray Series DataFrame

python 机器学习的基础库 numpy 和 pandas 定义了3种非常适合矩阵运算的数据结构:numpy.ndarray 、pandas.Series、pandas.DataFrame
它们之间有着很强的相关性

import pandas as pd
series = pd.Series([1, 2, 3, 4])
dataFrame = pd.DataFrame([[1, 2], [3, 4]])
  • ndarray: 这是最基础的类型,本质上就是一个n维矩阵
  • Series:它主要由两部分组成,index 和 values
    index是它的索引,在上面示例中的series由于没有指定index因此默认为[0, 1, 2, 3],索引可以是任意类型,不一定非是数字类型
    values是它存放的内容,是一个1维的ndarray
  • DataFrame: 可以将它看作是多列Series组成的,也可以看作是多行Series组成。或者可以看作主要由三部分组成,columns, index, values
    columns 是它的列名,默认也是数字升序,可以是任意类型
    index 是它的行名,默认数字升序,可以是任意类型
    values 是它存放的内容,是一个2维的ndarray

虽然这几种数据类型非常适合处理大数据,但是没能高效的使用它们会导致难以忍受的时间开销和空间开销

需要避免的几种操作

1. 大量的拼接操作

很多人想要在最终结果获得一个DataFrame类型的数据,就直接初始化一个空的DataFrame,每获得一条新数据,就对其进行拼接,获得最终的结果
实际上这3种数据类型非常不适合拼接

import pandas as pd
import time

time_start = time.time()
a = pd.Series()
for i in range(10000):
    a = pd.concat([a, pd.Series([i])])
time_end = time.time()
print("直接拼接", time_end - time_start)
time_start = time.time()
b = [0]
for i in range(10000):
    b.append(i)
pd.Series(b)
time_end = time.time()
print("list转换", time_end - time_start)

同样生成一个0到10000的Series
时间:

直接拼接 4.054653882980347
list转换 0.0050258636474609375

可以很明显的看到它们的时间差

2. 循环

我们用DataFrame 和 Series时候应该尽量用矩阵计算来代替循环操作
因为这几种数据类型本身就是为矩阵运算而设计的,矩阵运算的效率非常高
另外,Series的花式索引也是代替循环的一种好的方法:

meanMr = titanic.Age[titanic.Name == 'Mr'].mean()

这里的 titanic.Name == 'Mr' 会生成一个与titanic.Name 这个Series相同大小的Series,但是由True 和 False组成
titanic.Age中传入该索引,会留下索引中True那些项
所以 meanMr 获得的是 名字为Mr 的人的age的均值

推荐使用的操作

它们有大量内置的集合运算,可以求 交集,并集等,但是没有求补集的方法。
它们有大量用于查询的运算,类似SQL语句,有groupby语法,有sum average mean 等聚合函数

first_mention_id = first_df.groupby('mention_id')['label'].agg(['count', 'sum'])
first_mention_id = first_mention_id.index[(first_mention_id['count'] == 2) & (first_mention_id['sum'] == 1)]

first_df 是一个DataFrame 有 mention_id label 等列
这里的first_mention_id 就获得了有两个label 并且两个label加和为1的所有mention_id

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 199,393评论 5 467
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 83,790评论 2 376
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 146,391评论 0 330
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 53,703评论 1 270
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 62,613评论 5 359
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,003评论 1 275
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,507评论 3 390
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,158评论 0 254
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,300评论 1 294
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,256评论 2 317
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,274评论 1 328
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,984评论 3 316
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,569评论 3 303
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,662评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,899评论 1 255
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,268评论 2 345
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 41,840评论 2 339

推荐阅读更多精彩内容