如何正确获取nba投篮图数据?

时间:2021-01-08 10:56:38

标签: python nba-api

我想从 nba 站点获取数据以创建投篮图表。我从这个网站 https://datavizardry.com/2020/01/28/nba-shot-charts-part-1/?fbclid=IwAR3BheHQkSAmCJRr_z7ux1ygbspLjLdrvTRjWAVHOrr2BPvVh7jsIos_e9w 得到它,我的代码如下所示:

from nba_api.stats.endpoints import shotchartdetail
import json
import pandas as pd

response = shotchartdetail.ShotChartDetail(
    team_id=0,
    player_id=0,
    season_nullable='2001-02',
    season_type_all_star='Regular Season'
)

content = json.loads(response.get_json())

results = content['resultSets'][0]
headers = results['headers']
rows = results['rowSet']
df = pd.DataFrame(rows)
df.columns = headers

# write to csv file
df.to_csv(r'C:\Users\Comp\Desktop\nba_2001-02.csv', index=False)

它正在工作,但在本文中,当我从一开始下载示例数据时,数据中的作者也获得了关于错过投篮的信息,当我运行我的代码时,我只获得了投篮数据。是否也可以获得有关错过投篮的信息或 nba 不再共享此数据?

1 个答案:

答案 0 :(得分:0)

默认情况下包(可能自文章以来已更新)设置为仅返回得分的投篮,因此您需要将该参数更改为不是得分,而是投篮命中率,因为这将返回投篮和失手:

response = shotchartdetail.ShotChartDetail(
    team_id=0,
    player_id=0,
    season_nullable='2001-02',
    context_measure_simple = 'FGA', #<-- Default is 'PTS' and will only return made shots, but we want all shot attempts
    season_type_all_star='Regular Season'
)

另外,我会在数据框中再添加一件事,这样你就可以按名称而不是数字来列出列:

results = content['resultSets'][0]
headers = results['headers']
rows = results['rowSet']
df = pd.DataFrame(rows, columns=headers) #<-- add the columns parameter
df.columns = headers