Commit597552c

minjk-bl

committed

Fix data sampling code to prevent error on index issue

1 parent9e2b402 commit597552cCopy full SHA for 597552c

File tree

-6

lines changed

-6

lines changed

Lines changed: 6 additions & 6 deletions

Original file line number	Diff line number	Diff line change
`@@ -102,16 +102,16 @@ def _vp_sample(data, sample_cnt):`
`102`	`102`	`"""`
`103`	`103`	`Sampling data`
`104`	`104`	`"""`
`105`		`-dataType=type(data).__name__`
	`105`	`+data_type=type(data).__name__`
`106`	`106`	`sample_cnt=len(data)iflen(data)<sample_cntelsesample_cnt`
`107`	`107`
`108`		`-ifdataType=='DataFrame':`
	`108`	`+ifdata_type=='DataFrame':`
`109`	`109`	`returndata.sample(sample_cnt,random_state=0)`
`110`		`-elifdataType=='Series':`
`111`		`-returndata.sample(sample_cnt,random_state=0)`
`112`		`-elifdataType=='ndarray':`
	`110`	`+elifdata_type=='Series':`
	`111`	`+returndata.sample(sample_cnt,random_state=0).reset_index(drop=True)`
	`112`	`+elifdata_type=='ndarray':`
`113`	`113`	`returndata[_vp_np.random.choice(data.shape[0],sample_cnt,replace=False)]`
`114`		`-elifdataType=='list':`
	`114`	`+elifdata_type=='list':`
`115`	`115`	`return_vp_rd.choices(data,k=sample_cnt)`
`116`	`116`	`returndata`
`117`	`117`

Comments

(0)