指定开始和结束行号后，从Python的dataframe / csv文件中选择随机行（数据）？

Question

Using the sample() function I can get the random rows. Data set having 1000000 rows of data and I want to have a subset of 20000 rows. Importing random lines can be done through this solution

https://stackoverflow.com/a/22259008/8966221

reading a dataset

dataset = read_csv(file_path)

dataset_sub = dataset.sample(20000, random_state=1)

However I want to select random rows between row number 250000 to 750000 . Any possible solution in that regard?.

Answer 1

您可以做的是创建一个包含行号在250000到750000之间的行的DataFrame，然后从中选择20000个随机行。

dataset_sub = dataset.loc[250000:750000].sample(20000, random_state=1)

Answer 2

I think you need this:

dataset = read_csv(file_path)
dataset_sub = dataset.sample(random.randint(250000,750000), random_state=1)

Answer 3

I think the following code works:

import random
a=random.sample(range(250000,750000), 20000)
data=dataset.loc[a]