python - 如何从 pandas DataFrame 的子集进行采样？

Question

如何从第 0 行和长度 - 100 之间的 pandas DataFrame 中选择随机行？

类似于 start_state = self.market_data.sample(1)，只不过从任意行(甚至最后 100 行)中进行选择。

Answer 1

您可以使用DataFrame.iloc按位置选择。

如果需要过滤前 100 行:

self.market_data.iloc[:100].sample(1)

或者DataFrame.head :

self.market_data.head(100).sample(1)

<小时/>

如果要过滤最后 100 行之外的所有数据:

self.market_data.iloc[:-100].sample(1)