熊猫中 DataFrame.head() 的就地版本

Question

pandas 中是否有 df.head(max_rows) 的就地版本？

当数据框中的行数太多而无法处理时，我需要限制它们的数量。

目前我正在做df = df.head(10000000)，但我认为这是内存效率低下。

Answer 1

您可以使用pd.DataFrame.drop 进行就地操作：

n = 10000000
df.drop(df.index[n:], inplace=True)

但这可能无济于事。根据@unutbu's comment：

df.drop(..., inplace=True) 确实就地修改了df，但由于 方式inplace操作是在Pandas中实现的，没有真正的 这样做比更直接地重新分配到 变量名。我个人更喜欢返回值的函数 修改值的函数，因为前者是赋值 语法让修改内容一目了然。

这在Jeff's answer 中有进一步解释。

此外，请注意此方法不适用于重复索引。