非常有用的 Python 技巧

admin 2021年1月21日21:52:07评论204 views字数 2619阅读8分43秒阅读模式

非常有用的 Python 技巧

非常有用的 Python 技巧

函数连续调用


def add(x):
class AddNum(int):
def __call__(self, x):
return AddNum(self.numerator + x)
return AddNum(x)

print add(2)(3)(5)
# 10
print add(2)(3)(4)(5)(6)(7)
# 27

# javascript 版
var add = function(x){
var addNum = function(x){
return add(addNum + x);
};

addNum.toString = function(){
return x;
}
return addNum;
}

add(2)(3)(5)//10
add(2)(3)(4)(5)(6)(7)//27

默认值陷阱


>>> def evil(v=[]):
... v.append(1)
... print v
...
>>> evil()
[1]

>>> evil()
[1, 1]

读写csv文件


import csv

with open('data.csv', 'rb') as f:
reader = csv.reader(f)
for row in reader:
print row

# 向csv文件写入
import csv

with open( 'data.csv', 'wb') as f:
writer = csv.writer(f)
writer.writerow(['name', 'address', 'age']) # 单行写入
data = [
( 'xiaoming ','china','10'),
( 'Lily', 'USA', '12')]

writer.writerows(data) # 多行写入

数制转换


>>> int('1000', 2)
8

>>> int('A', 16)
10

格式化 json

echo'{"k": "v"}' | python-m json.tool

list 扁平化


list_ = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
[k for i in list_ for k in i] #[1, 2, 3, 4, 5, 6, 7, 8, 9]
import numpy as np
print np.r_[[1, 2, 3], [4, 5, 6], [7, 8, 9]]

import itertools
print list(itertools.chain(*[[1, 2, 3], [4, 5, 6], [7, 8, 9]]))
sum(list_, [])
flatten = lambda x: [y for l in x for y in flatten(l)] if type(x) is list else [x]
flatten(list_)

list 合并


>>> a = [1, 3, 5, 7, 9]
>>> b = [2, 3, 4, 5, 6]
>>> c = [5, 6, 7, 8, 9]
>>> list(set().union(a, b, c))
[1, 2, 3, 4, 5, 6, 7, 8, 9]

出现次数最多的 2 个字母


from collections import Counter
c = Counter('hello world')
print(c.most_common(2)) #[('l', 3), ('o', 2)]

谨慎使用

eval("__import__('os').system('rm -rf /')", {})

置换矩阵


matrix = [[1, 2, 3],[4, 5, 6]]
res = zip( *matrix ) # res = [(1, 4), (2, 5), (3, 6)]

列表推导


[item**2 for item in lst if item % 2]
map(lambda item: item ** 2, filter(lambda item: item % 2, lst))
>>> list(map(str, [1, 2, 3, 4, 5, 6, 7, 8, 9]))
['1', '2', '3', '4', '5', '6', '7', '8', '9']

排列组合


>>> for p in itertools.permutations([1, 2, 3, 4]):
... print ''.join(str(x) for x in p)
...
1234
1243
1324
1342
1423
1432
2134
2143
2314
2341
2413
2431
3124
3142
3214
3241
3412
3421
4123
4132
4213
4231
4312
4321

>>> for c in itertools.combinations([1, 2, 3, 4, 5], 3):
... print ''.join(str(x) for x in c)
...
123
124
125
134
135
145
234
235
245
345
>>> for c in itertools.combinations_with_replacement([1, 2, 3], 2):
... print ''.join(str(x) for x in c)
...
11
12
13
22
23
33
>>> for p in itertools.product([1, 2, 3], [4, 5]):
(1, 4)
(1, 5)
(2, 4)
(2, 5)
(3, 4)
(3, 5)

默认字典


>>> m = dict()
>>> m['a']
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
KeyError: 'a'
>>>
>>> m = collections.defaultdict(int)
>>> m['a']
0
>>> m['b']
0
>>> m = collections.defaultdict(str)
>>> m['a']
''
>>> m['b'] += 'a'
>>> m['b']
'a'
>>> m = collections.defaultdict(lambda: '[default value]')
>>> m['a']
'[default value]'
>>> m['b']
'[default value]'

反转字典


>>> m = {'a': 1, 'b': 2, 'c': 3, 'd': 4}
>>> m
{'d': 4, 'a': 1, 'b': 2, 'c': 3}
>>> {v: k for k, v in m.items()}
{1: 'a', 2: 'b', 3: 'c', 4: 'd'}

原文链接:https://mp.weixin.qq.com/s/ATvPzfLHwp0wEH5-tMW2cg

作者:苏生不惑  


非常有用的 Python 技巧

公众号专属福利 1 | 

| 2020全新专题实战教程限时1.99元 | 

|别获取课程邀请好友听课即可返现99%|

|如不分享也可识别海报中二维码单独购买  | 

非常有用的 Python 技巧
非常有用的 Python 技巧
非常有用的 Python 技巧
非常有用的 Python 技巧
非常有用的 Python 技巧
非常有用的 Python 技巧
非常有用的 Python 技巧

公众号专属福利 2 | 

| 长按识别免费领取实战手册| 

非常有用的 Python 技巧

非常有用的 Python 技巧

非常有用的 Python 技巧

非常有用的 Python 技巧

本文始发于微信公众号(马哥Linux运维):非常有用的 Python 技巧

  • 左青龙
  • 微信扫一扫
  • weinxin
  • 右白虎
  • 微信扫一扫
  • weinxin
admin
  • 本文由 发表于 2021年1月21日21:52:07
  • 转载请保留本文链接(CN-SEC中文网:感谢原作者辛苦付出):
                   非常有用的 Python 技巧http://cn-sec.com/archives/172397.html

发表评论

匿名网友 填写信息