python中高效刷题的扩展包

更多精彩内容,请关注【力扣简单题】

计数器

如果我们想统计列表或字符串中每个元素出现的次数,我们可以这样做:

def counter(l):
    count_dict = {}
    for e in l:
        if e in count_dict.keys():
            count_dict[e] += 1
        else:
            count_dict[e] = 1
    return count_dict

s = "apple"
c = counter(s)
print(c)
# {'a': 1, 'p': 2, 'l': 1, 'e': 1}

如果我们使用collection模块中的Counter,就可以快速进行统计:

from collections import Counter
s = "apple"
c = Counter(s)
print(c)
# Counter({'p': 2, 'a': 1, 'l': 1, 'e': 1})

列表也是可以统计的。

生成的Counter类可以看做一个字典,相当于字典的继承,支持加减操作,例如:

l1 = [7, 7, 7, 7, 8, 8, 8]
l2 = [7, 7, 7, 8, 8, 8, 8, 9]
c1, c2 = Counter(l1), Counter(l2)
print(c1+c2)
# Counter({7: 7, 8: 7, 9: 1})
print(c1-c2)
# Counter({7: 1})

排列组合

一个箱子里有写着1, 2, 3, 4的四张卡片,从中任意抽两张:

  1. 不考虑先后顺序,求所有可能的组合,需要使用combinations;

  2. 考虑先后顺序,且无放回的抽取,需要使用permutations;

  3. 考虑先后顺序,且有放回的抽取,需要使用product;

实验效果:

 from itertools import product, combinations, permutations
nums = [1, 2, 3, 4]
c = combinations(nums, r=2)
a = permutations(nums, r=2)
p = product(nums, repeat=2)
print(list(c))
# [(1, 2), (1, 3), (1, 4), (2, 3), (2, 4), (3, 4)]
print(list(a))
# [(1, 2), (1, 3), (1, 4), (2, 1), (2, 3), (2, 4), (3, 1), (3, 2), (3, 4), (4, 1), (4, 2), (4, 3)]
print(list(p))
# [(1, 1), (1, 2), (1, 3), (1, 4), (2, 1), (2, 2), (2, 3), (2, 4), (3, 1), (3, 2), (3, 3), (3, 4), (4, 1), (4, 2), (4, 3), (4, 4)]

数学计算

有时候,我们需要用到数学计算包,例如计算平方,对数等情况,数学计算包可以大大提高我们的编程效率。

from math import log, sqrt, pow
print(log(16, 2))
# 4
print(sqrt(2))
# 1.4142135623730951
print(pow(2, 6))
# 64.0

二分查找

帮助我们使用二分查找法快速定位元素在排序数组中的位置,或应该插入排序数组的位置。

import bisect
print(bisect.bisect_left([1, 2, 3, 4, 5], 2))
# 1
print(bisect.bisect_left([1, 2, 3, 4, 5], 3.5))
# 3
print(bisect.bisect_right([1, 2, 3, 4, 5], 3.5))
# 4
print(bisect.bisect_left([1, 2, 3, 4, 5], 5.5))
# 5
print(bisect.bisect_right([1, 2, 3, 4, 5], 5.5))
# 5
l = [1, 2, 3, 4, 5]
l.insert(bisect.bisect(l, 3.3), 3.3)
print(l)
# [1, 2, 3, 3.3, 4, 5]

灵活使用上述扩展包,会使我们的编码变得高效。

如有疑问或建议,欢迎评论区留言~

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

推荐阅读更多精彩内容

  • 一、Python简介和环境搭建以及pip的安装 4课时实验课主要内容 【Python简介】: Python 是一个...
    _小老虎_阅读 5,723评论 0 10
  • 今天任务被卡住了,有事只能到明天完成,所以能够按时下班。在楼梯遇到HR姐姐,跟她聊了一下,说到我们组的加班问题。看...
    UncleRiver阅读 209评论 0 1
  • 七绝 初冬山居见闻 雨打幽篁雀鸟惊, 流溪涧石拨瑶筝。 临川遥看云深处, 禅院依稀撞磬声。
    川伸阅读 242评论 2 7
  • 2019年3月21日 春分至,昼夜平分暖意浓。 可我这里却下雨了。 寄完给你们写好的明信片,长舒了一口气,像是完成...
    冰香oL阅读 450评论 3 2
  • 6月24日,中国广告协会第一次组团参加第43届戛纳国际广告节。 中国报协广告委员会三届一次会议通过《中国报纸广告行...
    许文超阅读 567评论 0 0