获取python numpy数组的列名

我有一个csv数据文件,其中包含一个表示列名的标题.

xy   wz  hi kq
0    10  5  6
1    2   4  7
2    5   2  6

我跑：

X = np.array(pd.read_csv('gbk_X_1.csv').values)

我想获取列名称：

['xy', 'wz', 'hi', 'kg']

我读了这个post,但解决方案为我提供了无.

解决方法:

使用以下代码：

import re

f = open('f.csv','r')

alllines = f.readlines()
columns = re.sub(' +',' ',alllines[0]) #delete extra space in one line
columns = columns.strip().split(',') #split using space

print(columns)

假设CSV文件是这样的：

xy   wz  hi kq
0    10  5  6
1    2   4  7
2    5   2  6