python - 如何使用 pandas 从 csv 读取多行到单个数据帧行

Question

我有一个文件，第一行有注释，后面是两行标题名称，第三行是索引名称。该文件如下所示:

# 3 5 <-- this is a comment indicating how many rows and column are matrix data
head1 head2 head3
head4 head5
idx1 idx2 idx3

1.1 1.2 1.3 
1.4 1.5
2.1 2.2 2.3
2.4 2.5
3.1 3.2 3.3
3.4 3.5

如何使用 pandas 读取文件以获得如下所示的数据框？

        head1 head2 head3 head4 head5
idx1    1.1   1.2   1.3   1.4   1.5
idx2    2.1   2.2   2.3   2.4   2.5
idx3    3.1   3.2   3.3   3.4   3.5

Answer 1

您可以指定 read_csv 的 skiprows 关键字为了创建一个包含所有 3 值线的数据框(通过跳过 2 值线)，然后创建另一个包含所有 2 值线的数据框。请注意，您可以使用 header 关键字指定标题行。

因此您可以将 csv 文件解析为两个不同的数据帧，您可以 concatenate稍后。

举个例子(假设三值线是偶数行，二值线是奇数行):

df3 = pd.read_csv(..., skiprows=lambda x: x%2 == 1)
df2 = pd.read_csv(..., skiprows=lambda x: x%2 == 0)

然后你可以使用concat为了将两个数据帧连接成一个:

df = pd.concat((df3, df2))