python 小技巧

把碰到的 python 小技巧记录下来

补全

a 是长度适中的字符串，现在要使它的长度是 n 的倍数。如果不够添加单字符 b。添加的个数小于等于n, 大于 0。

举例子：

如果 a 的长度刚好是 n 的倍数那么添加 n 个 b

a = 'akoder666'
b = '8'
n = 5

def fill_cha(ori, fill, num):
    length = (len(ori) + num) // num * num
    ori += b * (length - len(ori))
    return ori

fill_cha(a, b, n)

去掉字符串中的首尾的特殊字符

data = '\n\n\naaaa\n\n\n\n\v'
data.strip()

输出

aaaa

又一个简单的字符补全

a = '{:04}'.format(123)

输出

0123

numpy 合并两个数组

import numpy as np
a = np.array([1, 2, 3]) # shape(3, )
b = np.array([4, 5, 6]) # shape(3, )
c = np.c_[a, b]
# 这个的用法就是
# 只加 second axis，如果没有 second axis。默认为 1
# 所以
c.shape # (3, 1+1) => (3, 2)

去除两个集合相同的部分

a = {'123', 'qwe'}
b = {'123'}

a - b

输出

{'qwe'}

字符串按行分割

a = '123\n456\n789\n'
a.splitlines()

输出

['123', '456', '789']