写一段python程序:路径G:\drought\data\data1下有文件夹“20232300”和表格“station.xlsx”,遍历文件夹“20232300”中所有txt文件的名称,若表格第一列数据中有与之相同的,则复制该txt文件到文件夹“station”.
时间: 2024-09-10 18:21:15 浏览: 48
drought.io:干旱对美国影响的交互式可视化
以下是一段Python程序,用于完成您的要求:
```python
import os
import pandas as pd
# 定义路径
folder_path = r'G:\drought\data\data1\20232300'
xlsx_path = r'G:\drought\data\data1\station.xlsx'
target_folder = r'G:\drought\data\data1\station'
# 确保目标文件夹存在
if not os.path.exists(target_folder):
os.makedirs(target_folder)
# 读取表格中的第一列数据
df = pd.read_excel(xlsx_path)
first_column_data = df.iloc[:, 0].tolist()
# 遍历文件夹中的所有txt文件
for file_name in os.listdir(folder_path):
if file_name.endswith('.txt'):
# 检查文件名是否在表格的第一列数据中
if file_name in first_column_data:
# 构建完整的文件路径
file_path = os.path.join(folder_path, file_name)
# 复制文件到目标文件夹
os.system(f'copy {file_path} {target_folder}')
```
请确保您已经安装了`pandas`库,因为这段代码使用了pandas来读取Excel文件。如果还没有安装,可以通过`pip install pandas`来安装。
阅读全文