捕獲 python output 到變量

Question

我正在嘗試使pybaseball從玩家查找到數據提取自動化，但似乎無法通過將玩家 ID 設置為變量的第一步。

我正在使用以下代碼來提取玩家 ID：

from pybaseball import playerid_lookup

data = playerid_lookup("kershaw", "clayton", fuzzy=True)

上面幾行中的 output 是（精確復制/粘貼）：

  name_last name_first  key_mlbam key_retro  key_bbref  key_fangraphs  mlb_played_first  mlb_played_last
0   kershaw    clayton     477132  kersc001  kershcl01           2036            2008.0           2022.0

有沒有一種方法可以將 key_fangraphs 數字 (2036) 設置為來自這個 output 的變量？

Answer 1

由於pybaseball.playerid_lookup.playerid_lookup()方法返回pandas.DataFrame ，您應該能夠使用pandas.DataFrame.get()方法。 像data.get("key_fangraphs", default=-1)這樣的東西。 如果沒有“key_fangraphs”列，它將返回 -1。