筛选出包含如下站点名的文件名,站点名存储为一个numpy数组
时间: 2024-03-26 14:42:22 浏览: 62
03-python-numpy-简介-基础-数组-数组类型
假设你的站点名存储在一个名为`site_names`的numpy数组中,你可以使用以下代码来筛选出包含站点名的文件名:
```python
import numpy as np
import os
# 假设站点名存储在site_names中
site_names = np.array(['site1', 'site2', 'site3'])
# 设置文件夹路径
folder_path = '/path/to/your/folder'
# 获取文件夹中所有文件名
file_names = os.listdir(folder_path)
# 筛选出包含站点名的文件名
selected_file_names = []
for file_name in file_names:
for site_name in site_names:
if site_name in file_name:
selected_file_names.append(file_name)
break
# 输出筛选结果
print(selected_file_names)
```
上述代码中,`os.listdir`用于获取文件夹中的所有文件名,然后使用两层循环遍历站点名和文件名,如果站点名在文件名中出现,就将该文件名添加到`selected_file_names`列表中。最后,输出筛选结果。
阅读全文