age,gender,occupation,salary,address 22,Female,Student,2500,Beijing 32,Male,Engineer,8000,Shanghai 40,Male,Manager,12000,Guangzhou 28,Female,Teacher,5000,Beijing 35,Female,Doctor,15000,Shanghai 25,Male,Programmer,6000,Guangzhou 30,Female,Engineer,9000,Beijing 45,Male,Manager,18000,Shanghai如何把这个数据集写进CSV文件
时间: 2023-07-27 09:23:50 浏览: 40
人才数据集, csv 格式用于数据分析,或者向量数据库数据写入
将这个数据集写入 CSV 文件中可以使用 Python 中的 csv 模块。下面是一个示例代码:
```python
import csv
# 定义文件名
filename = 'data.csv'
# 定义数据
data = [
['age', 'gender', 'occupation', 'salary', 'address'],
[22, 'Female', 'Student', 2500, 'Beijing'],
[32, 'Male', 'Engineer', 8000, 'Shanghai'],
[40, 'Male', 'Manager', 12000, 'Guangzhou'],
[28, 'Female', 'Teacher', 5000, 'Beijing'],
[35, 'Female', 'Doctor', 15000, 'Shanghai'],
[25, 'Male', 'Programmer', 6000, 'Guangzhou'],
[30, 'Female', 'Engineer', 9000, 'Beijing'],
[45, 'Male', 'Manager', 18000, 'Shanghai'],
]
# 写入数据到文件中
with open(filename, 'w', newline='') as f:
writer = csv.writer(f)
writer.writerows(data)
```
在这个示例中,我们首先定义了文件名和数据。然后使用 `with` 语句打开文件,并创建一个 csv.writer 对象。使用 `writerows` 方法将数据写入文件中。最后,关闭文件。
需要注意的是,在打开文件的时候,需要指定 `newline=''` 参数,否则在 Windows 系统下可能会出现换行符问题。
阅读全文