获取python numpy数组的列名

Question

我有一个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']

我读了这篇文章，但是解决方案为我提供了None。

Answer 1

使用以下代码：

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