手把手教你如何简易封装自创的Python库,提高撸代码的效率
库的导入
写python的朋友第一行基本上应该都是import xxxyyy as xy
或者from xxxyyy import zzz as xyz
吧?咱不导入库的话代码也没法运行嘞。
我的moduleigh库
我至今也分享过大概10篇关于python的文章了,有些朋友可能也发现我有一个叫做moduleigh的库,不知道是个啥玩意儿,在pypi.org
上也没有,也没见别人用过这库。所以很显然这是我自己为了方便做的私人库:import moduleigh as ml
今天我就把它里面都写了啥给大家看看
- 首先呢,我也得导入别人的库嘞:
import pandas as pd
from sqlalchemy import create_engine
import pymysql
pymysql.install_as_MySQLdb()
- 以上的库都是啥我就不用做过多的解释了。
- 我要把我喜欢的Pandas输出显示设置提前预设好,代码如下:
def mypandasview():
# the usual pandas settings
pd.set_option('display.unicode.east_asian_width', True)
pd.set_option('display.unicode.ambiguous_as_wide', True)
pd.set_option('display.colheader_justify', 'right')
pd.set_option('expand_frame_repr', True)
pd.set_option('display.column_space', 50)
pd.set_option('display.max_columns', None)
pd.set_option('display.max_rows', None)
pd.set_option('display.width', 2000)
- 当我想调用的时候我就写一行代码执行就行了
ml.mypandasview()
- 下一个部分是我自己封装MySQL的代码:
def df_2_mysql(df, name, user='root', psw='root', ip='xx.xx.yy.yy:3306', db='stockdbleigh'):
"""
:param df: The data to pass
:param name: The name of the table in db
:param psw: Your password of your database
:param ip: Your IP
:param db: The name of your database
:param user: root
:return: None
"""
con = create_engine('mysql+pymysql://{}:{}@{}/{}'.format(user, psw, ip, db)) # mysql+pymysql的意思为:指定引擎为pymysql
df.to_sql(name, con, index=False, if_exists='replace', chunksize=None)
- 当我想把某个Dataframe数据传入到我的MySQL的时候我就输入
ml.df_2_mysql([Dataframe的赋值名称], [在mysql里该数据的命名])
- 最后的部分我都没有封装函数,就是一个
dictionary
:
mytoken = {
"Leigh": "570………………33ade",
"Healer": "f6b51………………cfd8",
"Leo": "c09c………………ee8c",
}
- 这是3个tushare的密钥,当我想用我的token的时候,就不用每次都要复制粘贴那一长串token了。然后分享代码的时候也不方便,把自己的token分享出去就BBQ了。
import tushare as ts
token = ml.mytoken['Leigh']
ts.set_token(token)
pro = ts.pro_api(token)
总结
- 这种方法是不是有点儿意思?大家都可以把一些经常重复敲打的代码打包再来进行调用,这样既可以节省时间也能提高书写效率。
- 但是这种方法也有个麻烦事儿就是你得把这个文件和导入它的文件放在一个文件夹下。我也尝试过进行全局设定导入那种方法,但是我失败了,囧。
- 看看以后有没有可能写一个关于可以完美的全局导入的方法分享给大家参考。
- 欢迎大家指点批评!!!
- 最后给大家普及一下发音:Leigh同Lee同Li同李。