­

太牛了!这个Python库竟然可以偷懒,和import说再见

  • 2019 年 12 月 17 日
  • 筆記
导读:本篇给大家介绍一个可以偷懒的python库:pyforest。

作者:xiaoyu,数据爱好者

来源:Python数据科学(ID:PyDataScience)

还记得入门Python数据分析时经常会import几个库,下面这几个可谓是入门学习时的四大护法,Python数据处理和可视化常会用的工具。

不知道大家有没有遇到过这样一个问题,每次重新开启一个建模流程或者分析过程时,会重新敲一遍import或者将之前的import代码copy进去。虽然已经用的滚花烂熟了,但是确耗费不必要的时间。

所以,你需要一个可以偷懒的python库:pyforest

01 一个示例

看到了吧,开头什么都没import,依然可以正常使用常用库。

02 安装

需要安装Python3.6以上的版本,然后终端运行:

pip install pyforest

链接:

https://github.com/8080labs/pyforest

03 使用方法

非常简单!!!

一行代码就可以解决。

from pyforest import *

如果使用Jupyter或IPython,甚至可以跳过此行,因为pyforest会将其自身添加到自动启动中。

而且,完成脚本后,可以通过以下方式导出所有导入语句:

active_imports()

通过这个指令,我们就可以看到脚本所涉及到的全部已经被我省略的库。

好了,你可能会问,如果需要的库不在pyforest中怎么办?

首先,pyforest支持大部分流行的数据科学库,比如pandas,numpy,matplotlib,seaborn,sklearn,tensorflow等等,以及常用的辅助库如os,sys,re,pickle等。

### Data Wrangling  pd = LazyImport("import pandas as pd")    np = LazyImport("import numpy as np")    dd = LazyImport("from dask import dataframe as dd")  SparkContext = LazyImport("from pyspark import SparkContext")    load_workbook = LazyImport("from openpyxl import load_workbook")    ### Data Visualization and Plotting  mpl = LazyImport("import matplotlib as mpl")  plt = LazyImport("import matplotlib.pyplot as plt")    sns = LazyImport("import seaborn as sns")    py = LazyImport("import plotly as py")  go = LazyImport("import plotly.graph_objs as go")  px = LazyImport("import plotly.express as px")    dash = LazyImport("import dash")    bokeh = LazyImport("import bokeh")    alt = LazyImport("import altair as alt")    pydot = LazyImport("import pydot")    # statistics  statistics = LazyImport("import statistics")    ### Machine Learning  sklearn = LazyImport("import sklearn")  OneHotEncoder = LazyImport("from sklearn.preprocessing import OneHotEncoder")  TSNE = LazyImport("from sklearn.manifold import TSNE")  train_test_split = LazyImport("from sklearn.model_selection import train_test_split")  svm = LazyImport("from sklearn import svm")  GradientBoostingClassifier = LazyImport(      "from sklearn.ensemble import GradientBoostingClassifier")  GradientBoostingRegressor = LazyImport(      "from sklearn.ensemble import GradientBoostingRegressor")  RandomForestClassifier = LazyImport(      "from sklearn.ensemble import RandomForestClassifier")  RandomForestRegressor = LazyImport("from sklearn.ensemble import RandomForestRegressor")    TfidfVectorizer = LazyImport(      "from sklearn.feature_extraction.text import TfidfVectorizer")    # TODO: add all the other most important sklearn objects  # TODO: add separate sections within machine learning viz. Classification, Regression, Error Functions, Clustering    # Deep Learning  tf = LazyImport("import tensorflow as tf")  keras = LazyImport("import keras")    # NLP  nltk = LazyImport("import nltk")  gensim = LazyImport("import gensim")  spacy = LazyImport("import spacy")  re = LazyImport("import re")    ### Helper  sys = LazyImport("import sys")  os = LazyImport("import os")  re = LazyImport("import re")  glob = LazyImport("import glob")  Path = LazyImport("from pathlib import Path")    pickle = LazyImport("import pickle")    dt = LazyImport("import datetime as dt")    tqdm = LazyImport("import tqdm")

其次,如果真的没有也没关系,pyforest支持向其中添加库。操作方法也很简单,找到pyforest库的user_imports.py文件,然后添加一个语句就好了,比如像下面这样:

#############################  ### User-specific imports ###  #############################  # You can save your own imports in ~/.pyforest/user_imports.py  # Please note: imports in ~/.pyforest/user_imports.py take precedence over the  # imports above.

所以我们可以根据平时自己的使用习惯添加一套属于自己的万能import,爽歪歪了。

有的同学可能还会问,所有的库添加进入运行速度会不会变慢? 答案是不会,因为只有你后面真正使用了到了pyforest里的包含的库,程序才会真正import,否则不会。 以上就是偷懒神器的介绍和使用方法,抓紧试试吧。