import pip


def install(package):
    if hasattr(pip, 'main'):
        pip.main(['install', package])
    else:
        pip._internal.main(['install', package])

# Example
if __name__ == '__main__':
    install('zipfile36')
import zipfile
zip = zipfile.ZipFile('chroma.zip')
zip.extractall()