import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
data=pd.read_csv('../kobe_data.csv')
# data=data.iloc[:1000,:]
data_x=np.array(data.loc[:,'loc_x'])
data_y=np.array(data.loc[:,'loc_y'])
#投篮类型
FG_2=data.loc[:,'shot_type']=='2PT Field Goal'
FG_3=data.loc[:,'shot_type']=='3PT Field Goal'
FG_2_data=data[FG_2]
FG_3_data=data[FG_3]
# print(FG_2_data)
# FG_2_data=FG_2_data.loc[:,'shot_made_flag']==1
#
# print(FG_2_data[FG_2_data])
#是否命中
FG_2_Y=FG_2_data.loc[:,'shot_made_flag']==1
FG_2_N=FG_2_data.loc[:,'shot_made_flag']==0
FG_3_Y=FG_3_data.loc[:,'shot_made_flag']==1
FG_3_N=FG_3_data.loc[:,'shot_made_flag']==0
#
# print(FG_2_Y)
data_FG_2_Y=FG_2_data[FG_2_Y]
data_FG_2_N=FG_2_data[FG_2_N]
#
data_FG_3_Y=FG_3_data[FG_3_Y]
data_FG_3_N=FG_3_data[FG_3_N]
# print(data_FG_2_Y)
#命中位置
data_FG_2_locx_Y=np.array(data_FG_2_Y.loc[:,'loc_x'])
data_FG_2_locy_Y=np.array(data_FG_2_Y.loc[:,'loc_y'])
data_FG_2_locx_N=np.array(data_FG_2_N.loc[:,'loc_x'])
data_FG_2_locy_N=np.array(data_FG_2_N.loc[:,'loc_y'])
#
data_FG_3_locx_Y=np.array(data_FG_3_Y.loc[:,'loc_x'])
data_FG_3_locy_Y=np.array(data_FG_3_Y.loc[:,'loc_y'])
data_FG_3_locx_N=np.array(data_FG_3_N.loc[:,'loc_x'])
data_FG_3_locy_N=np.array(data_FG_3_N.loc[:,'loc_y'])
# print(data_FG_2_locx_Y)
plt.subplot(121)
plt.scatter(data_FG_2_locx_Y,data_FG_2_locy_Y,color='green')
plt.title('2PT Field Goal')
# plt.scatter(data_FG_2_locx_N,data_FG_2_locy_N,color='red')
plt.subplot(122)
plt.scatter(data_FG_3_locx_Y,data_FG_3_locy_Y,color='green')
# plt.scatter(data_FG_3_locx_N,data_FG_3_locy_N,color='red')
plt.title('3PT Field Goal')
plt.show()
# print(data_FG_2_locx)