Generating a CSV file with headers using Pandas in Python 在Python中使用Pandas生成带有标头的CSV文件

Published on Aug. 22, 2023, 12:14 p.m.

您可以使用 Pandas 中的 to_csv 方法将 Pandas DataFrame 对象生成为 CSV 文件。以下是一个示例(摘自第六个搜索结果):

import pandas as pd

data = {'name': ['Bob', 'Jessica', 'Mary', 'John', 'Mel'], 'age': [21, 23, 18, 24, 20]}
df = pd.DataFrame(data)

df.to_csv('my_data.csv', index=False)

这将生成一个名为 my_data.csv 的 CSV 文件,并将 Pandas DataFrame 对象 df 写入该文件。index=False 参数可确保不将索引列写入 CSV 文件中。

Add header row

要在 Pandas 生成的 CSV 文件中添加标题行,您可以将 header 参数设置为一个包含标题名称的列表。以下是一个示例:

import pandas as pd

data = {'name': ['Bob', 'Jessica', 'Mary', 'John', 'Mel'], 'age': [21, 23, 18, 24, 20]}
df = pd.DataFrame(data)

# Add header row
header = ['Name', 'Age']

df.to_csv('my_data.csv', index=False, header=header)

这将生成一个名为 my_data.csv 的 CSV 文件,并将 Pandas DataFrame 对象 df 写入该文件。同时,header 参数被设置为一个包含两个字符串元素的列表。这样,输出的 CSV 文件会有一个名为 “Name” 的列和一个名为 “Age” 的列。

Tags: