Google Colab上で実行されているかどうかPythonで判定する

この記事は何

Google Colaboratory上で動かすことが前提のコードを書く場合、Colaboratory上で提供されていないライブラリはユーザーが自分でインストールする必要があります。たとえば以下のページのコードではdtreevizというライブラリを使用していて、Colab上で実行するためにはインストールが必要です。

【決定木のパラメータ - K_DM】

そのため、Colab上で実行されているか判定してColab上の場合のみインストールを実行できないか調べました。

解決方法

【Module: core.getipython — IPython 8.8.0 documentation】

get_ipython()を使うとipythonの実行環境を取得できます。 これを使うことでcolab上で実行されているかを判定できました。

コード例:

if str(get_ipython()).startswith("<google.colab."):
    !pip install dtreeviz

colabで実行されている場合のみ !pip install dtreeviz が実行されてdtreevizがインストールされます。