Google drive api 在google colab中保存csv测试数据？

我已经对测试数据进行了预测，但现在我想将测试数据的csv格式输出保存到google drive。有人能帮我吗

test_preds = predict_dl(test_dl, model_resnet18)

test_preds = [p.item() for p in test_preds]
test_preds

代码如下：

from google.colab import drive
drive.mount('/content/drive')
import numpy as np
import pandas as pd
test_preds = pd.DataFrame(test_preds).to_csv('/content/drive/My Drive/test_preds.csv')