pandas 写csv 加列名
时间: 2023-08-08 20:01:57 浏览: 444
在使用 Pandas 写入 CSV 文件时,可以通过两种方法为列添加列名。
方法一:使用 `header` 参数
可以在写入 CSV 文件时,通过设置 `header` 参数为 True,来添加列名。例如:
```python
import pandas as pd
# 创建数据
data = {'姓名': ['张三', '李四', '王五'],
'年龄': [20, 25, 30],
'性别': ['男', '女', '男']}
# 创建 DataFrame
df = pd.DataFrame(data)
# 写入 CSV 文件并添加列名
df.to_csv('data.csv', header=True, index=False)
```
这样会在生成的 CSV 文件的第一行添加列名。
方法二:使用 `columns` 参数
可以在创建 DataFrame 时,通过设置 `columns` 参数来指定列名。例如:
```python
import pandas as pd
# 创建数据
data = {'姓名': ['张三', '李四', '王五'],
'年龄': [20, 25, 30],
'性别': ['男', '女', '男']}
# 创建 DataFrame 并指定列名
df = pd.DataFrame(data, columns=['姓名', '年龄', '性别'])
# 写入 CSV 文件
df.to_csv('data.csv', index=False)
```
这样会在生成的 CSV 文件中添加指定的列名。
无论是使用哪种方法,最终都可以将指定的列名添加到写入的 CSV 文件中。
阅读全文