python - 获取 pandas 数据框中的标题行

Question

我有以下内容。文本文件:

line_1
line_2
line_3
Col_A    Col_B   Col_C
1.2      33       45
23       12       55

这是我的阅读方式:

df  = pandas.read_table(file_name, skiprows=3, sep=' ', skipinitialspace=True)

有什么方法可以只访问标题行？即'Col_A Col_B Col_C' 作为字符串？我不想使用 df.columns.values 因为它返回一个数组。我想要保留不同列之间的间距的标题行。

Answer 1

你可以这样做:

In [6]:
' '.join(df)

Out[6]:
'Col_A Col_B Col_C'

这是可行的，因为从 df 返回的可迭代对象是字符串列，因此您只需将它们与分隔符连接即可。

编辑

如果您想准确获取 header 存储的内容，则可以执行以下操作:

In [8]:
df = pd.read_table(io.StringIO(t), skiprows=3, header=None, nrows=1)
df

Out[8]:
                        0
0  Col_A    Col_B   Col_C

In [10]:
df.iloc[0][0]

Out[10]:
'Col_A    Col_B   Col_C'

因此，这没有指定分隔符，因此它将查找没有逗号的逗号，因此整个标题行被读取为单个列值，然后您可以通过如上所示对其进行索引来获取行值