用python将.csv文件的其中两列提取出来作文新的文件
时间: 2024-01-31 16:03:22 浏览: 76
.csv大文件分割插件
可以使用 Python 内置的 csv 模块来读取和写入 CSV 文件。下面是一个示例代码,用于从 CSV 文件中提取其中两列数据并将其写入新文件中:
```python
import csv
# 打开原始 CSV 文件和新文件
with open('input.csv', 'r') as csv_file, open('output.csv', 'w', newline='') as new_file:
# 创建 CSV 读取器和写入器
reader = csv.reader(csv_file)
writer = csv.writer(new_file)
# 循环读取原始文件的每一行
for row in reader:
# 获取需要的两列数据
column1 = row[0]
column2 = row[1]
# 将两列数据写入新文件中
writer.writerow([column1, column2])
```
请注意,这里假设原始 CSV 文件中的第一列和第二列分别是需要提取的两列数据。如果你的情况不同,请相应地更改代码中的 `row[0]` 和 `row[1]`。另外,这里将提取出的数据写入了名为 `output.csv` 的新文件中,你需要将其替换为你想要的文件名。
阅读全文