Google Colaboratory のファイルをVS Codeで編集する方法

Googleドライブから新規作成してブラウザで編集していますがVS Codeで編集したいです。 実際にやってみると、Googleドライブから作成したファイルをVS Codeで開くと部分的にうまく行っているように見えたりそうでなかったりします。 またVS Codeで拡張子が.ipynbのファイルを新規に作成して簡単なコードを記述して、GoogleドライブのColab Notebookフォルダに移して開こうとしてもエラーになり失敗します。 このような状態です。 何かうまい方法はあるのでしょうか。