4회 기출 유형(작업형2) 문제 풀이 과정 중에 질문드립니다.

pred = rf.predict_proba(test[:,1]) InvalidIndexError: (slice(None, None, None), 1)알려주신 내용을 입력하니,위와같은 에러가 발생하여서 원래대로 진행하였고,pred = rf.predict_proba(test) pd.DataFrame({'ID': test_id, 'Segmentation': pred[:,1]}).to_csv('000000.csv', index = False)이렇게 dataframe을 만들 때 'pred'에다가 [:,1]을 넣어 진행하니까 오류는 안뜨네요.그런데 모델링할 때는 pred[:,1] 없이 pred로만 진행해도 에러가 없었는데, 왜 오류가 생겼는지 이유를 잘 모르겠습니다. 왜 마지막 데이터프레임을 만들때만 [:,1]을 넣었는데 됐는지도 모르겠구요 ㅠㅠ