用PANDAS写代码。如果CSV的第一行以"apple)"."pear)"结尾,那么就分列。
时间: 2023-07-19 12:23:53 浏览: 72
pandas.read_csv参数详解(小结)
可以使用Pandas库读取CSV文件并进行处理。代码如下:
```python
import pandas as pd
df = pd.read_csv('filename.csv')
header = df.columns.tolist()
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}")
new_df = pd.DataFrame(columns=new_header)
for i in range(columns):
new_df[f"column_{i+1}"] = df.iloc[:, -2+i].tolist()
new_df.to_csv('new_filename.csv', index=False)
```
其中,`filename.csv`为原始CSV文件名,`new_filename.csv`为分列后的新文件名。如果原始CSV文件的第一行最后两列分别为"apple)"和"pear)",则会将其分列,并将结果写入新文件。新文件中的列名为"column_1"、"column_2"等。
阅读全文