python - 如何每隔一行读取 CSV 文件
问题描述
如何每 2 行从 CSV 文件中获取数据?
例如,如果我有一个看起来像这样的文件
0 1
0 23 34
1 45 45
2 78 16
3 110 78
4 48 14
5 76 23
6 55 33
7 12 13
8 18 76
如何迭代和提取每第二行以获得类似的内容并附加到新的数据帧中?
0 23 34
2 78 16
4 48 14
6 55 33
8 18 76
谢谢!
解决方案
使用 的skiprows
参数read_csv
:
要保持偶数行:
pd.read_csv('file.csv', skiprows=lambda x: (x != 0) and not x % 2)
要保留奇数行:
pd.read_csv('file.csv', skiprows=lambda x: x % 2)
请注意,标头包含在 中skiprows
,这就是x != 0
偶数示例中需要的原因。
例子:
In [1]: import pandas as pd
...: from io import StringIO
...:
...: data = """A,B
...: a,1
...: b,2
...: c,3
...: d,4
...: e,5
...: """
In [2]: pd.read_csv(StringIO(data))
Out[2]:
A B
0 a 1
1 b 2
2 c 3
3 d 4
4 e 5
In [3]: pd.read_csv(StringIO(data), skiprows=lambda x: (x != 0) and not x % 2)
Out[3]:
A B
0 a 1
1 c 3
2 e 5
In [4]: pd.read_csv(StringIO(data), skiprows=lambda x: x % 2)
Out[4]:
A B
0 b 2
1 d 4