python - pandas read_csv 并只保留某些行(python)

Question

我知道 skiprows 允许您传递一个列表，其中包含要跳过的行的索引。但是，我有要保留的行的索引。

假设我的 cvs 文件看起来像这样有数百万行:

  A B
0 1 2
1 3 4
2 5 6
3 7 8
4 9 0

我想加载的索引列表只有 2,3，所以

index_list = [2,3]

skiprows 函数的输入为 [0,1,4]。但是，我只有 [2,3] 可用。

我正在尝试类似的东西:

pd.read_csv(path, skiprows = ~index_list)

但没有运气..有什么建议吗？

谢谢，我感谢所有的帮助，

Answer 1

您可以在 skiprows 参数中传入一个 lambda 函数。例如:

rows_to_keep = [2,3]
pd.read_csv(path, skiprows = lambda x: x not in rows_to_keep)

您可以在文档 here 中阅读更多相关信息