使用 python/pandas 将数据框写入谷歌工作表

Question

我正在使用谷歌表格来保存共享项目的数据。使用 Google 的 Sheets API 我访问数据，在 python 中处理它，我试图在函数编写器中使用batchUpdate更新 Sheets 文件。

• 如果我将此函数数据作为列表传递，它会按预期工作。
• 如果我传递一个数据框（如我所愿），我会得到： TypeError: Object of type DataFrame is not JSON serializable
• 如果我使用 .to_json()，我会得到这个：

googleapiclient.errors.HttpError: <HttpError 400 当请求https://sheets.googleapis.com/v4/spreadsheets/XXX/values:batchUpdate?alt=json 时返回“‘data[0].values’处的无效值（类型。 googleapis.com/google.protobuf.ListValue), "{"0":{"0":1},"1":{"0":2},"2":{"0":3}," 3":{"0":4}}""。详细信息：“[{'@type'：'type.googleapis.com/google.rpc.BadRequest'，'fieldViolations'：[{'field'：'data[0].values'，'description'：'无效值在 'data[0].values' (type.googleapis.com/google.protobuf.ListValue), "{"0":{"0":1},"1":{"0":2}," 2":{"0":3},"3":{"0":4}}"'

任何指针将不胜感激。

import pickle
import os.path
import pandas as pd
from googleapiclient.discovery import build
from google_auth_oauthlib.flow import InstalledAppFlow
from google.auth.transport.requests import Request
from pprint import pprint

# If modifying these scopes, delete the file token.pickle.
SCOPES = ['https://www.googleapis.com/auth/spreadsheets']

# Spreadsheet ID: https://docs.google.com/spreadsheets/d/XXX/edit#gid=0
SPREADSHEET_ID = 'XXX'
RANGE_NAME = 'contacts'

def writer(df):
    service = build('sheets', 'v4', credentials=gsheet_api(SCOPES))
    sheet_name = 'contacts'
    data = [{'range' : sheet_name, 'values' : df}]
    batch_update_values_request_body = {
        'value_input_option': 'RAW',
        'data': data }

    request = service.spreadsheets().values().batchUpdate(spreadsheetId=SPREADSHEET_ID,
                                                          body=batch_update_values_request_body)
    response = request.execute()
    pprint(response)

df = [[1, 2, 3, 4]]
writer(df)

Answer 1

我相信你的目标和情况如下。

• 您想使用带有 Python 的 googleapis 将数据框放入 Google 电子表格。
• 您已经能够使用 Sheets API 获取和放置 Google 电子表格的值。

为此，这个答案怎么样？

改装要点：

• 我不确定数据框的值。因此，在此答案中，我想使用以下示例数据框解释修改点。

       A  B  C
    0  1  2  3
    1  4  5  6
    2  7  8  9
  

• 不幸的是，数据帧不能直接用于方法“spreadsheets.values.batchUpdate”的请求主体。所以在这种情况下，需要从数据帧转换为二维数组。为此，我使用了tolist().

当您使用示例数据框修改脚本时，它变成如下。

修改后的脚本：

从：

df = [[1, 2, 3, 4]]
writer(df)

到：

sampleValue = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
df = pd.DataFrame(sampleValue, columns=list('ABC'))
values = [df.columns.values.tolist()]
values.extend(df.values.tolist())
writer(values)

笔记：

• 如果不想包含标题行，请修改如下。

  • 从

      values = [df.columns.values.tolist()]
      values.extend(df.values.tolist())
    

  • 到

      values = df.values.tolist()
    

参考：

• 方法：spreadsheets.values.batchUpdate
• 列表