把一个csv文件中的第二列和第三列数据分别写入到两个.dat文件,文件名以csv文件的名字加上_1和_2。文件夹中有多个csv文件,都要这样处理
时间: 2024-03-20 20:44:28 浏览: 14
你可以使用 Python 中的 pandas 库来处理 CSV 文件并将数据写入.dat 文件。以下是一个示例代码:
```python
import pandas as pd
import os
# 获取当前文件夹中所有csv文件
csv_files = [f for f in os.listdir(".") if f.endswith(".csv")]
for csv_file in csv_files:
# 读取CSV文件
df = pd.read_csv(csv_file)
# 提取第二列和第三列数据
column_2 = df.iloc[:, 1]
column_3 = df.iloc[:, 2]
# 将数据写入.dat文件
file_name = os.path.splitext(csv_file)[0]
with open(file_name + "_1.dat", "w") as f:
f.write(column_2.to_string(index=False))
with open(file_name + "_2.dat", "w") as f:
f.write(column_3.to_string(index=False))
```
该代码会遍历当前文件夹中的所有以 `.csv` 结尾的文件,并将每个文件中的第二列和第三列数据分别写入以原文件名加上 `_1` 和 `_2` 的文件名中。注意要将文件名中的 `.csv` 后缀去掉。