csv列顺序 python脚本
时间: 2024-02-15 11:29:24 浏览: 114
python实现顺序表的简单代码
这是一个简单的Python脚本,它可以将CSV文件的列顺序更改为您想要的顺序。您需要将“input_file.csv”更改为您要更改顺序的文件的名称,并按照您想要的顺序更改“new_order”列表中的列名称。
```python
import csv
input_file = 'input_file.csv'
output_file = 'output_file.csv'
new_order = ['column1', 'column2', 'column3', 'column4']
with open(input_file, 'r') as csv_file:
reader = csv.DictReader(csv_file)
headers = reader.fieldnames
with open(output_file, 'w', newline='') as new_file:
writer = csv.DictWriter(new_file, fieldnames=new_order)
writer.writeheader()
for row in reader:
new_row = {}
for field in new_order:
new_row[field] = row[field]
writer.writerow(new_row)
```
在这个例子中,我们使用了Python的csv模块来读取和写入CSV文件。我们首先打开输入文件并使用csv.DictReader将其读入字典格式。我们使用reader.fieldnames获取文件的标题,然后打开输出文件并使用csv.DictWriter将其写入字典格式。我们使用writer.writeheader写入新标题。
接下来,我们遍历每行数据并创建一个新的字典new_row,其中只包含我们想要的列。我们使用writer.writerow将新行写入输出文件。
这个脚本可以轻松地修改以适应您的特定用例。只需更改input_file,output_file和new_order变量即可。
阅读全文