将D:\FILE\1012dataset中的所有csv文件转换成pkl文件并输出到D:\FILE\MILITARY
时间: 2024-10-14 18:11:13 浏览: 31
要将D:\FILE\1012dataset目录下的所有CSV文件转换为Pickle文件,并将结果保存到D:\FILE\MILITARY目录下,你可以使用Python编写脚本,结合`os`和`pandas`库。这里有一个基本的示例:
```python
import os
import pandas as pd
import pickle
# 指定源CSV文件夹和目标Pickle文件夹
src_folder = "D:\\FILE\\1012dataset"
dst_folder = "D:\\FILE\\MILITARY"
# 检查目标文件夹是否存在,不存在则创建
if not os.path.exists(dst_folder):
os.makedirs(dst_folder)
# 遍历源文件夹中的所有CSV文件
for filename in os.listdir(src_folder):
if filename.endswith('.csv'): # 确保只处理CSV文件
src_file = os.path.join(src_folder, filename)
dst_file = os.path.join(dst_folder, filename.split('.')[0] + '.pkl') # 构造目标Pickle文件名
# 读取CSV文件,转换为DataFrame,然后保存为Pickle文件
df = pd.read_csv(src_file)
with open(dst_file, 'wb') as handle:
pickle.dump(df, handle)
print(f"CSV文件已转换并保存到{dst_folder}.")
阅读全文