如果CSV的第一行以"apple)"."pear)"结尾,那么就分列。写出Python代码。
时间: 2023-07-19 19:23:53 浏览: 80
用python给csv里的数据排序的具体代码
5星 · 资源好评率100%
可以使用Python内置的csv模块,读取CSV文件并进行处理。代码如下:
```python
import csv
with open('filename.csv', 'r') as file:
csv_reader = csv.reader(file)
header = next(csv_reader)
if header[-1] == "apple)" and header[-2] == "pear)":
# 分列
columns = len(header) - 2
new_header = header[:-2]
for i in range(columns):
new_header.append(f"column_{i+1}")
# 写入新文件
with open('new_filename.csv', 'w', newline='') as new_file:
csv_writer = csv.writer(new_file)
csv_writer.writerow(new_header)
for row in csv_reader:
new_row = row[:-2]
for i in range(columns):
new_row.append(row[-2+i])
csv_writer.writerow(new_row)
```
其中,`filename.csv`为原始CSV文件名,`new_filename.csv`为分列后的新文件名。如果原始CSV文件的第一行最后两列分别为"apple)"和"pear)",则会将其分列,并将结果写入新文件。新文件中的列名为"column_1"、"column_2"等。
阅读全文