在Python中,你可以使用多种方法来创建和写入表头,具体取决于你想在哪个文件格式中使用表头(例如CSV、Excel等)。以下是使用不同方法创建表头的示例:
对于CSV文件
import csv
定义表头
header = ['标题', '租金', '付款方式', '可租年限', '房间数', '联系人', '电话', '地址']
写入表头到CSV文件
with open('data_list.csv', 'a', newline='') as f:
writer = csv.writer(f)
writer.writerow(header)
对于Excel文件
from openpyxl import Workbook
创建一个新的工作簿
wb = Workbook()
ws = wb.active
写入表头
headers = ['姓名', '年龄', '城市']
for col, header in enumerate(headers, 1):
ws.cell(row=1, column=col, value=header)
保存文件
wb.save('demo.xlsx')
使用Pandas库
import pandas as pd
假设表头位于第3行
header_line = 2
headers = open('data.csv', 'r').readlines()[header_line].strip().split(',')
使用pandas读取数据部分,忽略前面的行
df = pd.read_csv('data.csv', header=None, skiprows=range(header_line + 1))
df.columns = headers 设置正确的列名
print(df)
请根据你的具体需求选择合适的方法。