从PCollection（apachebeam、Python）中选择一些列

我有以下资料：

我只想从该PCollection中选择2列。我试着做：

def cut_data(data):
    return data[["WebSpeedRef", "WebSpeedAct"]]

data_min = data_json | 'min' >> beam.Map(cut_data)

---

但是我犯了一个错误。最简单的方法是什么。

什么类型的pcollection？熊猫？您得到了什么错误？您能提供您的错误吗？pcollection是什么类型的？熊猫？你有什么错误？你能提供你的错误吗？