如何使用Pandas获取并存储CSV文件的表头值?
存储CSV文件表头值的简便方法
当然有几种简单直接的方式,以下是主流编程语言里最常用的实现:
Python 常用方法
1. 基础csv.reader提取
直接读取文件第一行(表头行)存到变量即可:
import csv with open('your_file.csv', 'r') as f: reader = csv.reader(f) # 读取第一行作为表头 headers = next(reader) # headers就是存储表头的列表,可直接复用或保存 print(headers)
2. csv.DictReader直接获取fieldnames
用DictReader会自动识别表头,直接访问它的fieldnames属性就能拿到表头列表:
import csv with open('your_file.csv', 'r') as f: reader = csv.DictReader(f) headers = reader.fieldnames # 直接得到表头数据 print(headers)
3. Pandas快速提取(适合大数据场景)
如果用Pandas处理CSV,一行代码就能拿到表头:
import pandas as pd df = pd.read_csv('your_file.csv') headers = df.columns.tolist() # 存储或使用表头 print(headers)
JavaScript(Node.js)示例
用csv-parser包时,可通过headers事件获取表头:
const csv = require('csv-parser'); const fs = require('fs'); let headers; fs.createReadStream('your_file.csv') .pipe(csv()) .on('headers', (h) => { headers = h; console.log(headers); });
内容的提问来源于stack exchange,提问作者SushantIndupuru
相关产品推荐
相关产品推荐

