python - 获取 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']

我读了这个post但解决方案为我提供了 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